List of all items

Structs

• backend::Autodiff
• 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::audio::SpeechCommandsDataset
• data::dataset::audio::SpeechItem
• data::dataset::audio::SpeechItemRaw
• 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::transform::WindowsDataset
• data::dataset::transform::WindowsIterator
• 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
• lr_scheduler::step::StepLrScheduler
• lr_scheduler::step::StepLrSchedulerConfig
• module::Attribute
• module::ConstantRecord
• module::Content
• module::DisplaySettings
• module::Ignored
• module::Param
• module::ParamId
• module::Quantizer
• module::Reinitializer
• module::RunningState
• nn::BatchNorm
• nn::BatchNormConfig
• nn::BatchNormRecord
• nn::BatchNormRecordItem
• nn::BiLstm
• nn::BiLstmConfig
• nn::BiLstmRecord
• nn::BiLstmRecordItem
• 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::HardSigmoid
• nn::HardSigmoidConfig
• nn::InstanceNorm
• nn::InstanceNormConfig
• nn::InstanceNormRecord
• nn::InstanceNormRecordItem
• nn::LayerNorm
• nn::LayerNormConfig
• nn::LayerNormRecord
• nn::LayerNormRecordItem
• nn::LeakyRelu
• nn::LeakyReluConfig
• nn::Linear
• nn::LinearConfig
• nn::LinearRecord
• nn::LinearRecordItem
• nn::Lstm
• nn::LstmConfig
• nn::LstmRecord
• nn::LstmRecordItem
• nn::LstmState
• 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::Sigmoid
• nn::SwiGlu
• nn::SwiGluConfig
• nn::SwiGluRecord
• nn::SwiGluRecordItem
• nn::Tanh
• 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::Conv3d
• nn::conv::Conv3dConfig
• nn::conv::Conv3dRecord
• nn::conv::Conv3dRecordItem
• 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::conv::ConvTranspose3d
• nn::conv::ConvTranspose3dConfig
• nn::conv::ConvTranspose3dRecord
• nn::conv::ConvTranspose3dRecordItem
• nn::conv::DeformConv2d
• nn::conv::DeformConv2dConfig
• nn::conv::DeformConv2dRecord
• nn::conv::DeformConv2dRecordItem
• nn::gru::Gru
• nn::gru::GruConfig
• nn::gru::GruRecord
• nn::gru::GruRecordItem
• nn::interpolate::Interpolate1d
• nn::interpolate::Interpolate1dConfig
• nn::interpolate::Interpolate2d
• nn::interpolate::Interpolate2dConfig
• nn::loss::BinaryCrossEntropyLoss
• nn::loss::BinaryCrossEntropyLossConfig
• nn::loss::BinaryCrossEntropyLossRecord
• nn::loss::BinaryCrossEntropyLossRecordItem
• nn::loss::CosineEmbeddingLoss
• nn::loss::CosineEmbeddingLossConfig
• nn::loss::CrossEntropyLoss
• nn::loss::CrossEntropyLossConfig
• nn::loss::CrossEntropyLossRecord
• nn::loss::CrossEntropyLossRecordItem
• nn::loss::HuberLoss
• nn::loss::HuberLossConfig
• nn::loss::MseLoss
• nn::loss::PoissonNllLoss
• nn::loss::PoissonNllLossConfig
• nn::lstm::BiLstm
• nn::lstm::BiLstmConfig
• nn::lstm::BiLstmRecord
• nn::lstm::BiLstmRecordItem
• nn::lstm::Lstm
• nn::lstm::LstmConfig
• nn::lstm::LstmRecord
• nn::lstm::LstmRecordItem
• nn::lstm::LstmState
• 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::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::Float
• prelude::Int
• prelude::Shape
• prelude::Tensor
• prelude::TensorData
• record::BinBytesRecorder
• record::BinFileRecorder
• record::BinGzFileRecorder
• record::MaborMetadata
• record::MaborRecord
• record::MaborRecordNoItem
• 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::Bytes
• tensor::DimIter
• tensor::DistributionSampler
• tensor::Float
• tensor::Int
• tensor::PrintOptions
• tensor::Shape
• tensor::Slice
• tensor::T
• tensor::Tensor
• tensor::TensorData
• tensor::Tolerance
• tensor::Transaction
• tensor::backend::DeviceId
• tensor::bf16
• tensor::container::TensorContainer
• tensor::f16
• tensor::flex32
• tensor::grid::GridOptions
• tensor::ops::Conv2dBackward
• tensor::ops::Conv3dBackward
• tensor::ops::ConvOptions
• tensor::ops::ConvTransposeOptions
• tensor::ops::DeformConv2dBackward
• tensor::ops::DeformConvOptions
• tensor::ops::InterpolateBackward
• tensor::ops::InterpolateOptions
• tensor::ops::MaxPool1dBackward
• tensor::ops::MaxPool1dWithIndices
• tensor::ops::MaxPool2dBackward
• tensor::ops::MaxPool2dWithIndices
• tensor::ops::TransactionPrimitive
• tensor::ops::TransactionPrimitiveResult
• tensor::ops::UnfoldOptions
• tensor::quantization::CalibrationRange
• tensor::quantization::QParams
• tensor::quantization::QuantScheme
• tensor::quantization::QuantizationParametersPrimitive
• tensor::quantization::QuantizedBytes
• tensor::quantization::SymmetricQuantization
• train::ClassificationOutput
• train::FileApplicationLoggerInstaller
• train::Learner
• train::LearnerBuilder
• train::LearnerSummary
• train::MetricEarlyStoppingStrategy
• train::MetricEntry
• train::MetricSummary
• train::MultiDevicesTrainStep
• train::MultiLabelClassificationOutput
• train::RegressionOutput
• train::SummaryMetrics
• train::TrainEpoch
• train::TrainOutput
• train::TrainingInterrupter
• train::ValidEpoch
• train::checkpoint::AsyncCheckpointer
• train::checkpoint::ComposedCheckpointingStrategy
• train::checkpoint::ComposedCheckpointingStrategyBuilder
• train::checkpoint::FileCheckpointer
• train::checkpoint::KeepLastNCheckpoints
• train::checkpoint::MetricCheckpointingStrategy
• train::logger::AsyncLogger
• train::logger::FileLogger
• train::logger::FileMetricLogger
• train::logger::InMemoryLogger
• train::logger::InMemoryMetricLogger
• train::metric::AccuracyInput
• train::metric::AccuracyMetric
• train::metric::AurocInput
• train::metric::AurocMetric
• train::metric::ConfusionStatsInput
• train::metric::CpuMemory
• train::metric::CpuTemperature
• train::metric::CpuUse
• train::metric::CudaMetric
• train::metric::FBetaScoreMetric
• train::metric::HammingScore
• train::metric::HammingScoreInput
• train::metric::IterationSpeedMetric
• train::metric::LearningRateMetric
• train::metric::LossInput
• train::metric::LossMetric
• train::metric::MetricEntry
• train::metric::MetricMetadata
• train::metric::PrecisionMetric
• train::metric::RecallMetric
• train::metric::TopKAccuracyInput
• train::metric::TopKAccuracyMetric
• train::metric::state::FormatOptions
• train::metric::state::NumericMetricState
• train::metric::store::EventStoreClient
• train::metric::store::MetricsUpdate
• train::renderer::TrainingProgress
• train::renderer::tui::TuiMetricsRenderer
• train::train::MultiDevicesTrainStep

Enums

• config::ConfigError
• data::dataloader::Message
• data::dataset::ImporterError
• data::dataset::SqliteDatasetError
• data::dataset::audio::SpeechCommandClass
• 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::PaddingConfig3d
• nn::interpolate::InterpolateMode
• nn::loss::Reduction
• optim::record::AdaptorRecord
• optim::record::AdaptorRecordItem
• optim::record::AdaptorRecordItemV1
• optim::record::AdaptorRecordV1
• record::RecorderError
• serde::de::Unexpected
• tensor::DType
• tensor::DataError
• tensor::Distribution
• tensor::DistributionSamplerKind
• tensor::FloatDType
• tensor::Precision
• tensor::TensorPrimitive
• tensor::grid::GridIndexing
• tensor::grid::GridSparsity
• tensor::grid::IndexPos
• tensor::linalg::Norm
• tensor::ops::InterpolateMode
• tensor::quantization::Calibration
• tensor::quantization::QuantAccPrecision
• tensor::quantization::QuantInputType
• tensor::quantization::QuantLevel
• tensor::quantization::QuantMode
• tensor::quantization::QuantPropagation
• tensor::quantization::QuantizationStrategy
• train::StoppingCondition
• train::checkpoint::CheckpointerError
• train::checkpoint::CheckpointingAction
• train::metric::ClassReduction
• train::metric::NumericEntry
• train::metric::store::Aggregate
• train::metric::store::Direction
• train::metric::store::Event
• train::metric::store::Split
• train::renderer::MetricState

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
• data::dataset::transform::Window
• data::dataset::transform::Windows
• lr_scheduler::LrScheduler
• module::AutodiffModule
• module::Module
• module::ModuleDisplay
• module::ModuleDisplayDefault
• 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::AsIndex
• tensor::BasicAutodiffOps
• tensor::BasicOps
• tensor::BroadcastArgs
• tensor::Element
• tensor::ElementComparison
• tensor::ElementConversion
• tensor::ElementLimits
• tensor::ElementPrecision
• tensor::ElementRandom
• tensor::MovedimArgs
• tensor::Numeric
• tensor::RangeArg
• tensor::RangesArg
• tensor::ReshapeArgs
• tensor::TensorKind
• tensor::TensorMetadata
• tensor::backend::AutodiffBackend
• tensor::backend::Backend
• tensor::backend::DeviceOps
• tensor::cast::ToElement
• tensor::indexing::AsIndex
• tensor::ops::ActivationOps
• tensor::ops::BoolTensorOps
• tensor::ops::FloatTensorOps
• tensor::ops::IntTensorOps
• tensor::ops::ModuleOps
• tensor::ops::QTensorOps
• tensor::ops::TransactionOps
• tensor::quantization::QTensorPrimitive
• tensor::quantization::Quantization
• train::ApplicationLoggerInstaller
• train::EarlyStoppingStrategy
• train::TrainStep
• train::ValidStep
• train::checkpoint::Checkpointer
• train::checkpoint::CheckpointingStrategy
• train::logger::Logger
• train::logger::LoggerBackend
• train::logger::MetricLogger
• train::metric::Adaptor
• train::metric::ItemLazy
• train::metric::Metric
• train::metric::Numeric
• train::metric::store::EventStore
• train::renderer::MetricsRenderer

Macros

• constant
• prelude::s
• serde::forward_to_deserialize_any
• tensor::dequant_op_flow
• tensor::dequant_op_quant
• tensor::make_element
• tensor::s

Derive Macros

• config::Config
• module::Module
• prelude::Config
• prelude::Module
• record::Record
• serde::Deserialize
• serde::Serialize

Functions

• config::config_to_json
• data::dataloader::split::split_dataloader
• module::extract_type_name
• module::list_param_ids
• nn::attention::generate_autoregressive_mask
• nn::attention::generate_padding_mask
• nn::generate_sinusoids
• tensor::activation::gelu
• tensor::activation::hard_sigmoid
• 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::softmin
• tensor::activation::softplus
• tensor::activation::tanh
• tensor::argsort
• tensor::argwhere_data
• tensor::cartesian_grid
• tensor::check_closeness
• tensor::grid::meshgrid
• tensor::grid::meshgrid_stack
• tensor::indexing::canonicalize_dim
• tensor::indexing::canonicalize_index
• tensor::indexing::wrap_index
• tensor::linalg::cosine_similarity
• tensor::linalg::l0_norm
• tensor::linalg::l1_norm
• tensor::linalg::l2_norm
• tensor::linalg::lp_norm
• tensor::linalg::max_abs_norm
• tensor::linalg::min_abs_norm
• tensor::linalg::vector_norm
• tensor::linalg::vector_normalize
• 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::conv3d
• tensor::module::conv_transpose1d
• tensor::module::conv_transpose2d
• tensor::module::conv_transpose3d
• tensor::module::deform_conv2d
• tensor::module::embedding
• tensor::module::interpolate
• tensor::module::linear
• tensor::module::max_pool1d
• tensor::module::max_pool1d_with_indices
• tensor::module::max_pool2d
• tensor::module::max_pool2d_with_indices
• tensor::module::unfold4d
• tensor::ops::binary_ops_shape
• tensor::ops::conv::calculate_conv_output_size
• tensor::ops::conv::calculate_conv_output_sizes
• tensor::ops::conv::calculate_conv_padding
• tensor::ops::conv::calculate_conv_transpose_output_size
• tensor::ops::conv::calculate_pool_output_size
• tensor::quantization::pack_i8s_to_u32s
• tensor::quantization::unpack_u32s_to_i8s
• tensor::read_sync
• tensor::set_print_options
• tensor::sort
• tensor::sort_with_indices
• tensor::try_read_sync
• train::metric::format_float

Type Aliases

• LearningRate
• 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::IntElem
• tensor::ops::IntTensor
• tensor::ops::QuantizedTensor
• tensor::quantization::QuantizationParameters

Constants

• tensor::DEFAULT_ATOL
• tensor::DEFAULT_RTOL
• tensor::linalg::DEFAULT_EPSILON