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::state::BackwardStates
• 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::CandleQTensor
• backend::candle::CandleTensor
• backend::candle::CudaDevice
• backend::candle::MetalDevice
• backend::candle::PrecisionBridge
• backend::cuda_jit::CudaDevice
• backend::hip_jit::HipDevice
• backend::libtorch::LibTorch
• backend::libtorch::PrecisionBridge
• backend::libtorch::TchQTensor
• backend::libtorch::TchShape
• backend::libtorch::TchTensor
• backend::ndarray::NdArray
• backend::ndarray::NdArrayQTensor
• backend::ndarray::NdArrayTensor
• backend::ndarray::PrecisionBridge
• backend::wgpu::AutoGraphicsApi
• backend::wgpu::CubeDim
• 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::WgpuResource
• backend::wgpu::WgpuRuntime
• backend::wgpu::WgpuServer
• backend::wgpu::WgpuStorage
• backend::wgpu::WgslCompiler
• 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::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
• module::Attribute
• module::ConstantRecord
• module::Content
• module::DisplaySettings
• module::Ignored
• module::Param
• module::ParamId
• module::Quantizer
• 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::CrossEntropyLoss
• nn::loss::CrossEntropyLossConfig
• nn::loss::CrossEntropyLossRecord
• nn::loss::CrossEntropyLossRecordItem
• nn::loss::HuberLoss
• nn::loss::HuberLossConfig
• nn::loss::MseLoss
• 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::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::Float
• prelude::Int
• prelude::Shape
• prelude::Tensor
• prelude::TensorData
• 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::TensorData
• tensor::backend::DeviceId
• tensor::bf16
• tensor::container::TensorContainer
• tensor::f16
• 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::UnfoldOptions
• tensor::quantization::AffineQuantization
• tensor::quantization::CalibrationRange
• tensor::quantization::MinMaxCalibration
• tensor::quantization::QuantizationParameters
• tensor::quantization::QuantizationParametersPrimitive
• tensor::quantization::SymmetricQuantization
• tensor::repr::AdaptiveAvgPool1dBackwardDescription
• tensor::repr::AdaptiveAvgPool1dDescription
• tensor::repr::AdaptiveAvgPool2dBackwardDescription
• tensor::repr::AdaptiveAvgPool2dDescription
• tensor::repr::AvgPool1dBackwardDescription
• tensor::repr::AvgPool1dDescription
• tensor::repr::AvgPool2dBackwardDescription
• tensor::repr::AvgPool2dDescription
• tensor::repr::BinaryOperationDescription
• tensor::repr::CatOperationDescription
• tensor::repr::ClampOperationDescription
• tensor::repr::Conv1dDescription
• tensor::repr::Conv1dOptionsDescription
• tensor::repr::Conv2dDescription
• tensor::repr::Conv2dOptionsDescription
• tensor::repr::Conv3dDescription
• tensor::repr::Conv3dOptionsDescription
• tensor::repr::ConvTranspose1dDescription
• tensor::repr::ConvTranspose1dOptionsDescription
• tensor::repr::ConvTranspose2dDescription
• tensor::repr::ConvTranspose2dOptionsDescription
• tensor::repr::ConvTranspose3dDescription
• tensor::repr::ConvTranspose3dOptionsDescription
• tensor::repr::DeformConv2dBackwardDescription
• tensor::repr::DeformConv2dDescription
• tensor::repr::DeformableConv2dOptionsDescription
• tensor::repr::DequantizeOperationDescription
• tensor::repr::EmbeddingBackwardDescription
• tensor::repr::EmbeddingDescription
• tensor::repr::ExpandDescription
• tensor::repr::ExpandOperationDescription
• tensor::repr::FlipOperationDescription
• tensor::repr::GatherOperationDescription
• tensor::repr::HandleContainer
• tensor::repr::InterpolateBackwardDescription
• tensor::repr::InterpolateDescription
• tensor::repr::InterpolateOptionsDescription
• tensor::repr::MaskFillOperationDescription
• tensor::repr::MaskWhereOperationDescription
• tensor::repr::MaxPool1dDescription
• tensor::repr::MaxPool1dWithIndicesBackwardDescription
• tensor::repr::MaxPool1dWithIndicesDescription
• tensor::repr::MaxPool2dDescription
• tensor::repr::MaxPool2dWithIndicesBackwardDescription
• tensor::repr::MaxPool2dWithIndicesDescription
• tensor::repr::PermuteOperationDescription
• tensor::repr::QuantizationParametersDescription
• tensor::repr::QuantizeOperationDescription
• tensor::repr::QuantizedKind
• tensor::repr::QuantizedTensorDescription
• tensor::repr::RandomOperationDescription
• tensor::repr::ReduceDimWithIndicesDescription
• tensor::repr::RepeatDimOperationDescription
• tensor::repr::ReshapeDescription
• tensor::repr::ScalarOperationDescription
• tensor::repr::ScatterOperationDescription
• tensor::repr::SelectAssignOperationDescription
• tensor::repr::SelectOperationDescription
• tensor::repr::SliceAssignOperationDescription
• tensor::repr::SliceOperationDescription
• tensor::repr::SwapDimsDescription
• tensor::repr::TensorDescription
• tensor::repr::TensorHandle
• tensor::repr::TensorId
• tensor::repr::UnaryOperationDescription
• 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::CpuMemory
• train::metric::CpuTemperature
• train::metric::CpuUse
• train::metric::CudaMetric
• train::metric::HammingScore
• train::metric::HammingScoreInput
• train::metric::LearningRateMetric
• train::metric::LossInput
• train::metric::LossMetric
• train::metric::MetricEntry
• train::metric::MetricMetadata
• 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::train::MultiDevicesTrainStep

Enums

• backend::autodiff::ops::OpsKind
• backend::candle::CandleDevice
• backend::libtorch::LibTorchDevice
• backend::libtorch::Storage
• backend::ndarray::NdArrayDevice
• backend::wgpu::MemoryConfiguration
• 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::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::Precision
• tensor::TensorPrimitive
• tensor::ops::InterpolateMode
• tensor::quantization::QuantizationScheme
• tensor::quantization::QuantizationStrategy
• tensor::quantization::QuantizationType
• tensor::repr::BaseOperationDescription
• tensor::repr::BoolOperationDescription
• tensor::repr::FloatOperationDescription
• tensor::repr::Handle
• tensor::repr::HandleKind
• tensor::repr::IntOperationDescription
• tensor::repr::InterpolateModeDescription
• tensor::repr::ModuleOperationDescription
• tensor::repr::NumericOperationDescription
• tensor::repr::OperationDescription
• tensor::repr::TensorStatus
• train::StoppingCondition
• train::checkpoint::CheckpointerError
• train::checkpoint::CheckpointingAction
• train::metric::NumericEntry
• train::metric::store::Aggregate
• train::metric::store::Direction
• train::metric::store::Event
• train::metric::store::Split
• train::renderer::MetricState

Traits

• backend::autodiff::checkpoint::retro_forward::RetroForward
• backend::autodiff::checkpoint::strategy::CheckpointStrategy
• backend::autodiff::ops::Backward
• backend::libtorch::QuantElement
• 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
• 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::BasicAutodiffOps
• tensor::BasicOps
• tensor::BroadcastArgs
• tensor::Element
• tensor::ElementComparison
• tensor::ElementConversion
• tensor::ElementPrecision
• tensor::ElementRandom
• tensor::MovedimArgs
• tensor::Numeric
• tensor::RangesArg
• tensor::ReshapeArgs
• tensor::TensorKind
• tensor::backend::AutodiffBackend
• tensor::backend::Backend
• tensor::backend::BackendBridge
• tensor::backend::DeviceOps
• tensor::cast::ToElement
• tensor::ops::ActivationOps
• tensor::ops::BoolTensorOps
• tensor::ops::FloatTensorOps
• tensor::ops::IntTensorOps
• tensor::ops::ModuleOps
• tensor::ops::QTensorOps
• tensor::quantization::Calibration
• tensor::quantization::QTensorPrimitive
• tensor::quantization::Quantization
• tensor::repr::ReprBackend
• 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::Metric
• train::metric::Numeric
• train::metric::store::EventStore
• train::renderer::MetricsRenderer

Macros

• backend::autodiff::retro_binary
• backend::autodiff::retro_unary
• backend::autodiff::retro_unary_scalar
• backend::ndarray::reshape
• backend::ndarray::to_typed_dims
• constant
• serde::forward_to_deserialize_any
• tensor::dequant_op_quant
• 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::create_client
• backend::wgpu::create_wgpu_setup
• backend::wgpu::init_async
• backend::wgpu::init_existing_device
• backend::wgpu::init_memory_management
• backend::wgpu::init_sync
• backend::wgpu::select_device
• config::config_to_json
• 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::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::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::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::binary_ops_shape
• 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::read_sync
• tensor::set_print_options
• tensor::sort
• tensor::sort_with_indices
• tensor::try_read_sync
• train::metric::format_float

Type Aliases

• LearningRate
• backend::CudaJit
• backend::Wgpu
• backend::autodiff::grads::GradID
• backend::cuda_jit::Cuda
• backend::hip_jit::Hip
• backend::libtorch::StorageRef
• backend::wgpu::Wgpu
• backend::wgpu::Wgsl
• 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
• tensor::ops::QuantizedTensor