List of all items

Structs

• monitor::EpochMetrics
• monitor::MonitorBuilder
• monitor::TrainingMonitor
• prelude::Adam
• prelude::AdamW
• prelude::AddNoise
• prelude::AdversarialTrainer
• prelude::AvgPool2d
• prelude::BCELoss
• prelude::BasicBPETokenizer
• prelude::BatchNorm1d
• prelude::BatchNorm2d
• prelude::Bert
• prelude::BertConfig
• prelude::BertForMaskedLM
• prelude::BertForSequenceClassification
• prelude::Bottleneck
• prelude::BottleneckAnalyzer
• prelude::CenterCrop
• prelude::CharTokenizer
• prelude::CompiledFunction
• prelude::ComputeProfiler
• prelude::Conv2d
• prelude::CosineAnnealingLR
• prelude::CrossEntropyLoss
• prelude::DataLoader
• prelude::DistributedDataParallel
• prelude::Dropout
• prelude::EarlyStopping
• prelude::Embedding
• prelude::ExponentialLR
• prelude::GELU
• prelude::GPT2
• prelude::GPT2Config
• prelude::GPT2LMHead
• prelude::GRU
• prelude::GenerationConfig
• prelude::Graph
• prelude::ImageNormalize
• prelude::JitCompiler
• prelude::JitOptimizer
• prelude::L1Loss
• prelude::LSTM
• prelude::LanguageModelDataset
• prelude::LayerNorm
• prelude::LeNet
• prelude::LeakyReLU
• prelude::Linear
• prelude::MFCC
• prelude::MSELoss
• prelude::MaxPool2d
• prelude::MelSpectrogram
• prelude::MemoryProfiler
• prelude::MultiHeadAttention
• prelude::NormalizeAudio
• prelude::Parameter
• prelude::ProcessGroup
• prelude::ProfileGuard
• prelude::ProfileReport
• prelude::Profiler
• prelude::ProgressLogger
• prelude::RMSprop
• prelude::RNN
• prelude::RandomHorizontalFlip
• prelude::RandomSampler
• prelude::ReLU
• prelude::Resample
• prelude::Resize
• prelude::SGD
• prelude::Sequential
• prelude::SequentialSampler
• prelude::SiLU
• prelude::Sigmoid
• prelude::SimpleCNN
• prelude::Softmax
• prelude::StepLR
• prelude::SyntheticCIFAR
• prelude::SyntheticCommandDataset
• prelude::SyntheticMNIST
• prelude::SyntheticMusicDataset
• prelude::SyntheticSentimentDataset
• prelude::Tanh
• prelude::Tensor
• prelude::TextDataset
• prelude::TextGenerator
• prelude::TimelineProfiler
• prelude::TracedValue
• prelude::TrainingConfig
• prelude::TrainingHistory
• prelude::TrainingMetrics
• prelude::Variable
• prelude::Vocab
• prelude::WhitespaceTokenizer
• prelude::World
• prelude::no_grad::EnableGradGuard
• prelude::no_grad::InferenceModeGuard
• prelude::no_grad::NoGradGuard
• prelude::trace::TracedValue
• prelude::trace::Tracer

Enums

• prelude::DType
• prelude::Device
• prelude::Error

Traits

• prelude::Callback
• prelude::Dataset
• prelude::LRScheduler
• prelude::Module
• prelude::Optimizer
• prelude::Tokenizer
• prelude::Transform

Functions

• features
• prelude::all_reduce_mean
• prelude::all_reduce_sum
• prelude::barrier
• prelude::broadcast
• prelude::no_grad
• prelude::no_grad::enable_grad
• prelude::no_grad::inference_mode
• prelude::no_grad::is_grad_enabled
• prelude::no_grad::is_inference_mode
• prelude::no_grad::no_grad
• prelude::no_grad::set_grad_enabled_context
• prelude::trace
• prelude::trace::trace
• version

Type Aliases

• prelude::DDP
• prelude::Result