List of all items

Structs

• autograd::AdaptiveAvgPool2dBackward
• autograd::AddBackward
• autograd::AddBiasBackward
• autograd::AddChannelBiasBackward
• autograd::BatchNorm2dBackward
• autograd::BmmBackward
• autograd::BroadcastAddBackward
• autograd::BroadcastMulBackward
• autograd::BroadcastSubBackward
• autograd::CastBackward
• autograd::CatBackward
• autograd::CrossEntropyBackward
• autograd::CustomBackwardOp
• autograd::DropoutBackward
• autograd::EmbeddingBackward
• autograd::FlattenBackward
• autograd::GeluBackward
• autograd::GradientStore
• autograd::Im2ColBackward
• autograd::LayerNormBackward
• autograd::LeakyReluBackward
• autograd::MatmulBackward
• autograd::MaxPool2dBackward
• autograd::MseLossBackward
• autograd::MulBackward
• autograd::ReluBackward
• autograd::ReshapeBackward
• autograd::SigmoidBackward
• autograd::SliceBatchBackward
• autograd::SliceRangeBackward
• autograd::SoftmaxBackward
• autograd::StackBackward
• autograd::SubBackward
• autograd::TanhBackward
• autograd::Tape
• autograd::TapeEntry
• autograd::TransposeBackward
• autograd::VersionSnapshot
• autograd::context::NoGradGuard
• backend::CpuBackend
• data::DataItem
• data::DataLoader
• data::record::IndexEntry
• data::record::RecordDataset
• data::record::RecordWriter
• nn::AdaptiveAvgPool2d
• nn::BatchNorm2d
• nn::Conv2d
• nn::ConvTranspose2d
• nn::Dropout
• nn::Embedding
• nn::Flatten
• nn::LayerNorm
• nn::Linear
• nn::MaxPool2d
• nn::Parameter
• nn::attention::MultiheadAttention
• nn::transformer::TransformerBlock
• optim::Adam
• optim::AdamW
• optim::SGD
• optim::scheduler::CosineAnnealingLR
• optim::scheduler::StepLR
• tensor::FenceId
• tensor::GradId
• tensor::Layout
• tensor::OpId
• tensor::ParamId
• tensor::StorageHandle
• tensor::StorageInner
• tensor::Tensor
• tensor::TensorMeta
• tensor::WeakStorageHandle
• tensor::broadcast::BroadcastInfo
• train::Trainer

Enums

• autograd::AutogradError
• autograd::BackwardOp
• data::DataLoaderIter
• tensor::AutogradState
• tensor::DType

Traits

• autograd::CustomBackward
• backend::Backend
• data::Dataset
• nn::Module
• optim::Optimizer
• optim::scheduler::LRScheduler

Derive Macros

• nn::Module

Functions

• autograd::backward
• autograd::backward_with_grad
• autograd::context::install_tape
• autograd::context::no_grad
• autograd::context::take_tape
• nn::attention::scaled_dot_product_attention
• nn::cross_entropy_loss
• nn::gelu
• nn::leaky_relu
• nn::load_safetensors
• nn::mse_loss
• nn::relu
• nn::save_safetensors
• nn::sigmoid
• nn::tanh
• optim::clip::clip_grad_norm_
• tensor::broadcast::broadcast_shape
• tensor::broadcast::broadcast_strides
• tensor::broadcast::cpu_broadcast_binary
• tensor::broadcast::cpu_reduce_sum
• tensor::broadcast::suffix_products
• tensor::cat
• tensor::stack

Type Aliases

• tensor::DataReadGuard
• tensor::DataWriteGuard