List of all items

Structs

• backend::Autodiff
• backend::Candle
• backend::Fusion
• backend::LibTorch
• backend::NdArray
• backend::Wgpu
• data::dataloader::BatchDataLoader
• data::dataloader::DataLoaderBuilder
• data::dataloader::DatasetIterator
• data::dataloader::FixBatchStrategy
• data::dataloader::MultiThreadDataLoader
• data::dataloader::Progress
• data::dataset::DatasetIterator
• data::dataset::HuggingfaceDatasetLoader
• data::dataset::InMemDataset
• data::dataset::SqliteDataset
• data::dataset::SqliteDatasetStorage
• data::dataset::SqliteDatasetWriter
• data::dataset::source::huggingface::HuggingfaceDatasetLoader
• data::dataset::transform::ComposedDataset
• data::dataset::transform::MapperDataset
• data::dataset::transform::PartialDataset
• data::dataset::transform::SamplerDataset
• data::dataset::transform::ShuffledDataset
• data::dataset::vision::MNISTDataset
• data::dataset::vision::MNISTItem
• lr_scheduler::constant::ConstantLr
• lr_scheduler::noam::NoamLrScheduler
• lr_scheduler::noam::NoamLrSchedulerConfig
• module::ConstantRecord
• module::Param
• module::ParamId
• module::RunningState
• nn::BatchNorm
• nn::BatchNormConfig
• nn::BatchNormRecord
• nn::BatchNormRecordItem
• nn::Dropout
• nn::DropoutConfig
• nn::Embedding
• nn::EmbeddingConfig
• nn::EmbeddingRecord
• nn::EmbeddingRecordItem
• nn::GELU
• nn::GateController
• nn::GateControllerRecord
• nn::GateControllerRecordItem
• nn::GroupNorm
• nn::GroupNormConfig
• nn::GroupNormRecord
• nn::GroupNormRecordItem
• nn::LayerNorm
• nn::LayerNormConfig
• nn::LayerNormRecord
• nn::LayerNormRecordItem
• nn::Linear
• nn::LinearConfig
• nn::LinearRecord
• nn::LinearRecordItem
• nn::Lstm
• nn::LstmConfig
• nn::LstmRecord
• nn::LstmRecordItem
• nn::PositionalEncoding
• nn::PositionalEncodingConfig
• nn::PositionalEncodingRecord
• nn::PositionalEncodingRecordItem
• nn::ReLU
• nn::Unfold4d
• nn::Unfold4dConfig
• nn::attention::GeneratePaddingMask
• nn::attention::MhaCache
• nn::attention::MhaInput
• nn::attention::MhaOutput
• nn::attention::MultiHeadAttention
• nn::attention::MultiHeadAttentionConfig
• nn::attention::MultiHeadAttentionRecord
• nn::attention::MultiHeadAttentionRecordItem
• nn::cache::TensorCache
• nn::conv::Conv1d
• nn::conv::Conv1dConfig
• nn::conv::Conv1dRecord
• nn::conv::Conv1dRecordItem
• nn::conv::Conv2d
• nn::conv::Conv2dConfig
• nn::conv::Conv2dRecord
• nn::conv::Conv2dRecordItem
• nn::conv::ConvTranspose1d
• nn::conv::ConvTranspose1dConfig
• nn::conv::ConvTranspose1dRecord
• nn::conv::ConvTranspose1dRecordItem
• nn::conv::ConvTranspose2d
• nn::conv::ConvTranspose2dConfig
• nn::conv::ConvTranspose2dRecord
• nn::conv::ConvTranspose2dRecordItem
• nn::gru::Gru
• nn::gru::GruConfig
• nn::gru::GruRecord
• nn::gru::GruRecordItem
• nn::loss::BinaryCrossEntropyLoss
• nn::loss::BinaryCrossEntropyLossConfig
• nn::loss::BinaryCrossEntropyLossRecord
• nn::loss::BinaryCrossEntropyLossRecordItem
• nn::loss::CrossEntropyLoss
• nn::loss::CrossEntropyLossConfig
• nn::loss::CrossEntropyLossRecord
• nn::loss::CrossEntropyLossRecordItem
• nn::loss::MSELoss
• nn::lstm::Lstm
• nn::lstm::LstmConfig
• nn::lstm::LstmRecord
• nn::lstm::LstmRecordItem
• nn::pool::AdaptiveAvgPool1d
• nn::pool::AdaptiveAvgPool1dConfig
• nn::pool::AdaptiveAvgPool2d
• nn::pool::AdaptiveAvgPool2dConfig
• nn::pool::AvgPool1d
• nn::pool::AvgPool1dConfig
• nn::pool::AvgPool2d
• nn::pool::AvgPool2dConfig
• nn::pool::MaxPool1d
• nn::pool::MaxPool1dConfig
• nn::pool::MaxPool2d
• nn::pool::MaxPool2dConfig
• nn::transformer::PositionWiseFeedForward
• nn::transformer::PositionWiseFeedForwardConfig
• nn::transformer::PositionWiseFeedForwardRecord
• nn::transformer::PositionWiseFeedForwardRecordItem
• nn::transformer::TransformerDecoder
• nn::transformer::TransformerDecoderAutoregressiveCache
• nn::transformer::TransformerDecoderConfig
• nn::transformer::TransformerDecoderInput
• nn::transformer::TransformerDecoderLayer
• nn::transformer::TransformerDecoderLayerRecord
• nn::transformer::TransformerDecoderLayerRecordItem
• nn::transformer::TransformerDecoderRecord
• nn::transformer::TransformerDecoderRecordItem
• nn::transformer::TransformerEncoder
• nn::transformer::TransformerEncoderAutoregressiveCache
• nn::transformer::TransformerEncoderConfig
• nn::transformer::TransformerEncoderInput
• nn::transformer::TransformerEncoderLayer
• nn::transformer::TransformerEncoderLayerRecord
• nn::transformer::TransformerEncoderLayerRecordItem
• nn::transformer::TransformerEncoderRecord
• nn::transformer::TransformerEncoderRecordItem
• optim::AdaGrad
• optim::AdaGradConfig
• optim::AdaGradState
• optim::AdaGradStateItem
• optim::Adam
• optim::AdamConfig
• optim::AdamState
• optim::AdamStateItem
• optim::AdamW
• optim::AdamWConfig
• optim::AdamWState
• optim::AdamWStateItem
• optim::AdaptiveMomentumState
• optim::AdaptiveMomentumStateItem
• optim::AdaptiveMomentumWState
• optim::AdaptiveMomentumWStateItem
• optim::CenteredState
• optim::CenteredStateItem
• optim::GradientsAccumulator
• optim::GradientsParams
• optim::LRDecayState
• optim::LRDecayStateItem
• optim::RMSProp
• optim::RMSPropConfig
• optim::RMSPropMomentum
• optim::RMSPropMomentumState
• optim::RMSPropMomentumStateItem
• optim::RMSPropState
• optim::RMSPropStateItem
• optim::Sgd
• optim::SgdConfig
• optim::SgdState
• optim::SgdStateItem
• optim::SquareAvgState
• optim::SquareAvgStateItem
• optim::adaptor::OptimizerAdaptor
• optim::decay::WeightDecay
• optim::decay::WeightDecayConfig
• optim::decay::WeightDecayState
• optim::decay::WeightDecayStateItem
• optim::momentum::Momentum
• optim::momentum::MomentumConfig
• optim::momentum::MomentumState
• optim::momentum::MomentumStateItem
• record::BinBytesRecorder
• record::BinFileRecorder
• record::BinGzFileRecorder
• record::BurnMetadata
• record::BurnRecord
• record::BurnRecordNoItem
• record::DoublePrecisionSettings
• record::FullPrecisionSettings
• record::HalfPrecisionSettings
• record::JsonGzFileRecorder
• record::NamedMpkBytesRecorder
• record::NamedMpkFileRecorder
• record::NamedMpkGzFileRecorder
• record::ParamSerde
• record::PrettyJsonFileRecorder
• tensor::Bool
• tensor::Data
• tensor::DataSerialize
• tensor::DimIter
• tensor::DistributionSampler
• tensor::Float
• tensor::Int
• tensor::PrintOptions
• tensor::Shape
• tensor::T
• tensor::Tensor
• tensor::bf16
• tensor::container::TensorContainer
• tensor::f16
• tensor::ops::Conv1dBackward
• tensor::ops::Conv2dBackward
• tensor::ops::ConvOptions
• tensor::ops::ConvTransposeOptions
• tensor::ops::MaxPool1dBackward
• tensor::ops::MaxPool1dWithIndices
• tensor::ops::MaxPool2dBackward
• tensor::ops::MaxPool2dWithIndices
• tensor::ops::UnfoldOptions

Enums

• config::ConfigError
• data::dataloader::Message
• data::dataset::ImporterError
• data::dataset::SqliteDatasetError
• data::dataset::source::huggingface::ImporterError
• grad_clipping::GradientClipping
• grad_clipping::GradientClippingConfig
• nn::Initializer
• nn::PaddingConfig1d
• nn::PaddingConfig2d
• nn::loss::Reduction
• optim::record::AdaptorRecord
• optim::record::AdaptorRecordItem
• optim::record::AdaptorRecordItemV1
• optim::record::AdaptorRecordV1
• record::RecorderError
• tensor::Distribution
• tensor::DistributionSamplerKind
• tensor::Precision
• tensor::Reader

Traits

• config::Config
• data::dataloader::BatchStrategy
• data::dataloader::DataLoader
• data::dataloader::DataLoaderIterator
• data::dataloader::Dataset
• data::dataloader::batcher::Batcher
• data::dataset::Dataset
• data::dataset::transform::Mapper
• lr_scheduler::LrScheduler
• module::AutodiffModule
• module::Module
• module::ModuleMapper
• module::ModuleVisitor
• optim::Optimizer
• optim::SimpleOptimizer
• record::BytesRecorder
• record::FileRecorder
• record::PrecisionSettings
• record::Record
• record::Recorder
• tensor::BasicAutodiffOps
• tensor::BasicOps
• tensor::Element
• tensor::ElementConversion
• tensor::ElementPrecision
• tensor::ElementRandom
• tensor::Numeric
• tensor::ReshapeArgs
• tensor::TensorKind
• tensor::backend::AutodiffBackend
• tensor::backend::Backend
• tensor::ops::ActivationOps
• tensor::ops::BoolTensorOps
• tensor::ops::FloatTensorOps
• tensor::ops::IntTensorOps
• tensor::ops::ModuleOps

Macros

• constant
• module::constant
• tensor::make_element

Derive Macros

• config::Config
• module::Module
• record::Record

Functions

• config::config_to_json
• module::list_param_ids
• nn::attention::generate_autoregressive_mask
• nn::attention::generate_padding_mask
• nn::generate_sinusoids
• tensor::activation::gelu
• tensor::activation::log_sigmoid
• tensor::activation::log_softmax
• tensor::activation::mish
• tensor::activation::quiet_softmax
• tensor::activation::relu
• tensor::activation::sigmoid
• tensor::activation::silu
• tensor::activation::softmax
• tensor::activation::softplus
• tensor::activation::tanh
• tensor::chunk
• tensor::loss::cross_entropy_with_logits
• tensor::module::adaptive_avg_pool1d
• tensor::module::adaptive_avg_pool2d
• tensor::module::avg_pool1d
• tensor::module::avg_pool2d
• tensor::module::conv1d
• tensor::module::conv2d
• tensor::module::conv_transpose1d
• tensor::module::conv_transpose2d
• tensor::module::embedding
• tensor::module::max_pool1d
• tensor::module::max_pool1d_with_indices
• tensor::module::max_pool2d
• tensor::module::max_pool2d_with_indices
• tensor::module::unfold4d
• tensor::narrow
• tensor::ops::conv::calculate_conv_output_size
• tensor::ops::conv::calculate_conv_padding
• tensor::ops::conv::calculate_conv_transpose_output_size
• tensor::ops::conv::calculate_pool_output_size
• tensor::set_print_options

Type Aliases

• LearningRate
• data::dataset::Result
• module::Devices
• record::CompactRecorder
• record::DebugRecordSettings
• record::DefaultFileRecorder
• record::DefaultRecorder
• record::NoStdInferenceRecorder
• record::NoStdTrainingRecorder
• record::SensitiveCompactRecorder
• tensor::Device
• tensor::ops::BoolTensor
• tensor::ops::Device
• tensor::ops::FloatElem
• tensor::ops::FloatTensor
• tensor::ops::FullPrecisionBackend
• tensor::ops::IntElem
• tensor::ops::IntTensor