List of all items
Structs
- backend::Autodiff
- backend::Candle
- backend::LibTorch
- backend::NdArray
- backend::autodiff::Autodiff
- backend::autodiff::AutodiffBridge
- backend::autodiff::NodeID
- backend::autodiff::checkpoint::base::Checkpointer
- backend::autodiff::checkpoint::strategy::BalancedCheckpointing
- backend::autodiff::checkpoint::strategy::NoCheckpointing
- backend::autodiff::grads::Gradients
- backend::autodiff::ops::ComputePropertyDone
- backend::autodiff::ops::Init
- backend::autodiff::ops::MemoryBound
- backend::autodiff::ops::MemoryBoundRetroForward
- backend::autodiff::ops::Ops
- backend::autodiff::ops::OpsPrep
- backend::autodiff::ops::Tracked
- backend::autodiff::ops::UnTracked
- backend::candle::Candle
- backend::candle::CandleTensor
- backend::candle::PrecisionBridge
- backend::libtorch::LibTorch
- backend::libtorch::PrecisionBridge
- backend::libtorch::TchShape
- backend::libtorch::TchTensor
- backend::ndarray::NdArray
- backend::ndarray::NdArrayTensor
- backend::ndarray::PrecisionBridge
- backend::wgpu::AutoGraphicsApi
- backend::wgpu::Dx12
- backend::wgpu::JitBackend
- backend::wgpu::JitTensor
- backend::wgpu::Metal
- backend::wgpu::OpenGl
- backend::wgpu::RuntimeOptions
- backend::wgpu::Vulkan
- backend::wgpu::WebGpu
- backend::wgpu::WgpuRuntime
- backend::wgpu::WorkGroup
- 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::BoundingBox
- data::dataset::vision::ImageDatasetItem
- data::dataset::vision::ImageFolderDataset
- data::dataset::vision::MnistDataset
- data::dataset::vision::MnistItem
- data::dataset::vision::SegmentationMask
- lr_scheduler::constant::ConstantLr
- lr_scheduler::cosine::CosineAnnealingLrScheduler
- lr_scheduler::cosine::CosineAnnealingLrSchedulerConfig
- lr_scheduler::exponential::ExponentialLrScheduler
- lr_scheduler::exponential::ExponentialLrSchedulerConfig
- lr_scheduler::linear::LinearLrScheduler
- lr_scheduler::linear::LinearLrSchedulerConfig
- 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::GateController
- nn::GateControllerRecord
- nn::GateControllerRecordItem
- nn::Gelu
- nn::GroupNorm
- nn::GroupNormConfig
- nn::GroupNormRecord
- nn::GroupNormRecordItem
- nn::InstanceNorm
- nn::InstanceNormConfig
- nn::InstanceNormRecord
- nn::InstanceNormRecordItem
- nn::LayerNorm
- nn::LayerNormConfig
- nn::LayerNormRecord
- nn::LayerNormRecordItem
- nn::LeakyRelu
- nn::LeakyReluConfig
- nn::LeakyReluRecord
- nn::LeakyReluRecordItem
- nn::Linear
- nn::LinearConfig
- nn::LinearRecord
- nn::LinearRecordItem
- nn::Lstm
- nn::LstmConfig
- nn::LstmRecord
- nn::LstmRecordItem
- nn::PRelu
- nn::PReluConfig
- nn::PReluRecord
- nn::PReluRecordItem
- nn::PositionalEncoding
- nn::PositionalEncodingConfig
- nn::PositionalEncodingRecord
- nn::PositionalEncodingRecordItem
- nn::Relu
- nn::RmsNorm
- nn::RmsNormConfig
- nn::RmsNormRecord
- nn::RmsNormRecordItem
- nn::RotaryEncoding
- nn::RotaryEncodingConfig
- nn::RotaryEncodingRecord
- nn::RotaryEncodingRecordItem
- nn::SwiGlu
- nn::SwiGluConfig
- nn::SwiGluRecord
- nn::SwiGluRecordItem
- 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::HuberLoss
- nn::loss::HuberLossConfig
- nn::loss::HuberLossRecord
- nn::loss::HuberLossRecordItem
- 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
- prelude::Bool
- prelude::Data
- prelude::Float
- prelude::Int
- prelude::Shape
- prelude::Tensor
- 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
- serde::de::IgnoredAny
- serde::de::value::BoolDeserializer
- serde::de::value::BorrowedBytesDeserializer
- serde::de::value::BorrowedStrDeserializer
- serde::de::value::BytesDeserializer
- serde::de::value::CharDeserializer
- serde::de::value::CowStrDeserializer
- serde::de::value::EnumAccessDeserializer
- serde::de::value::Error
- serde::de::value::F32Deserializer
- serde::de::value::F64Deserializer
- serde::de::value::I128Deserializer
- serde::de::value::I16Deserializer
- serde::de::value::I32Deserializer
- serde::de::value::I64Deserializer
- serde::de::value::I8Deserializer
- serde::de::value::IsizeDeserializer
- serde::de::value::MapAccessDeserializer
- serde::de::value::MapDeserializer
- serde::de::value::SeqAccessDeserializer
- serde::de::value::SeqDeserializer
- serde::de::value::StrDeserializer
- serde::de::value::StringDeserializer
- serde::de::value::U128Deserializer
- serde::de::value::U16Deserializer
- serde::de::value::U32Deserializer
- serde::de::value::U64Deserializer
- serde::de::value::U8Deserializer
- serde::de::value::UnitDeserializer
- serde::de::value::UsizeDeserializer
- serde::ser::Impossible
- 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::InterpolateBackward
- tensor::ops::InterpolateOptions
- tensor::ops::MaxPool1dBackward
- tensor::ops::MaxPool1dWithIndices
- tensor::ops::MaxPool2dBackward
- tensor::ops::MaxPool2dWithIndices
- tensor::ops::UnfoldOptions
Enums
- backend::autodiff::ops::OpsKind
- backend::candle::CandleDevice
- backend::libtorch::LibTorchDevice
- backend::libtorch::Storage
- backend::ndarray::NdArrayDevice
- backend::wgpu::WgpuDevice
- config::ConfigError
- data::dataloader::Message
- data::dataset::ImporterError
- data::dataset::SqliteDatasetError
- data::dataset::source::huggingface::ImporterError
- data::dataset::vision::Annotation
- data::dataset::vision::ImageLoaderError
- data::dataset::vision::PixelDepth
- 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
- serde::de::Unexpected
- tensor::Distribution
- tensor::DistributionSamplerKind
- tensor::Precision
- tensor::Reader
- tensor::ops::InterpolateMode
Traits
- backend::autodiff::checkpoint::strategy::CheckpointStrategy
- backend::autodiff::ops::Backward
- backend::libtorch::TchElement
- backend::ndarray::FloatNdArrayElement
- backend::wgpu::FloatElement
- backend::wgpu::GraphicsApi
- backend::wgpu::IntElement
- backend::wgpu::WgpuElement
- config::Config
- data::dataloader::BatchStrategy
- data::dataloader::DataLoader
- data::dataloader::DataLoaderIterator
- data::dataloader::Dataset
- data::dataloader::DynDataLoader
- data::dataloader::batcher::Batcher
- data::dataloader::batcher::DynBatcher
- data::dataset::Dataset
- data::dataset::transform::Mapper
- lr_scheduler::LrScheduler
- module::AutodiffModule
- module::Module
- module::ModuleMapper
- module::ModuleVisitor
- module::Parameter
- optim::Optimizer
- optim::SimpleOptimizer
- prelude::Backend
- prelude::Config
- prelude::ElementConversion
- prelude::Module
- record::BytesRecorder
- record::FileRecorder
- record::PrecisionSettings
- record::Record
- record::Recorder
- serde::Deserialize
- serde::Deserializer
- serde::Serialize
- serde::Serializer
- serde::de::Deserialize
- serde::de::DeserializeOwned
- serde::de::DeserializeSeed
- serde::de::Deserializer
- serde::de::EnumAccess
- serde::de::Error
- serde::de::Expected
- serde::de::IntoDeserializer
- serde::de::MapAccess
- serde::de::SeqAccess
- serde::de::StdError
- serde::de::VariantAccess
- serde::de::Visitor
- serde::ser::Error
- serde::ser::Serialize
- serde::ser::SerializeMap
- serde::ser::SerializeSeq
- serde::ser::SerializeStruct
- serde::ser::SerializeStructVariant
- serde::ser::SerializeTuple
- serde::ser::SerializeTupleStruct
- serde::ser::SerializeTupleVariant
- serde::ser::Serializer
- serde::ser::StdError
- tensor::BasicAutodiffOps
- tensor::BasicOps
- tensor::BroadcastArgs
- tensor::Element
- tensor::ElementComparison
- tensor::ElementConversion
- tensor::ElementPrecision
- tensor::ElementRandom
- tensor::Numeric
- tensor::ReshapeArgs
- tensor::TensorKind
- tensor::backend::AutodiffBackend
- tensor::backend::Backend
- tensor::backend::BackendBridge
- tensor::ops::ActivationOps
- tensor::ops::BoolTensorOps
- tensor::ops::FloatTensorOps
- tensor::ops::IntTensorOps
- tensor::ops::ModuleOps
Macros
- backend::autodiff::retro_binary
- backend::autodiff::retro_unary
- backend::autodiff::retro_unary_scalar
- backend::ndarray::iter_par
- backend::ndarray::iter_range_par
- backend::ndarray::reshape
- backend::ndarray::run_par
- backend::ndarray::to_typed_dims
- constant
- serde::forward_to_deserialize_any
- tensor::make_element
Derive Macros
- config::Config
- module::Module
- prelude::Config
- prelude::Module
- record::Record
- serde::Deserialize
- serde::Serialize
Functions
- backend::autodiff::ops::binary
- backend::autodiff::ops::broadcast_shape
- backend::autodiff::ops::unary
- backend::autodiff::ops::unary_different_backend
- backend::wgpu::init_async
- backend::wgpu::init_sync
- backend::wgpu::select_device
- 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::leaky_relu
- tensor::activation::log_sigmoid
- tensor::activation::log_softmax
- tensor::activation::mish
- tensor::activation::prelu
- tensor::activation::quiet_softmax
- tensor::activation::relu
- tensor::activation::sigmoid
- tensor::activation::silu
- tensor::activation::softmax
- tensor::activation::softplus
- tensor::activation::tanh
- tensor::argsort
- tensor::argwhere
- 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::interpolate
- 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
- tensor::sort
- tensor::sort_with_indices
Type Aliases
- LearningRate
- backend::Wgpu
- backend::autodiff::grads::GradID
- backend::libtorch::StorageRef
- backend::wgpu::Wgpu
- data::dataset::Result
- module::Devices
- prelude::Device
- 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