List of all items
Structs
- CpuBackend
- CpuDevice
- CudaBackend
- CudaDevice
- GradStore
- Layout
- Shape
- Tensor
- TensorId
- checkpoint::TrainingCheckpoint
- distributed::DataParallel
- distributed::LossScaleConfig
- distributed::MixedPrecisionMetrics
- distributed::MixedPrecisionTrainer
- distributed::ParallelTrainer
- distributed::PipelineParallel
- distributed::PipelineStage
- exec::EpochLog
- exec::ExecResult
- exec::Executor
- exec::RuntimeConfig
- exec::TrainResult
- exec::Trainer
- exec::jit::CompileStats
- exec::jit::CompiledGraph
- exec::jit::JitExecutor
- exec::jit::JitResult
- exec::jit::MemoryPlan
- exec::jit::SlotRef
- exec::jit::ValueLifetime
- ir::Error
- ir::IrProgram
- ir::Program
- ir::ast::Arg
- ir::ast::AssertStmt
- ir::ast::CheckBlock
- ir::ast::ConfigBlock
- ir::ast::CustomOpBlock
- ir::ast::ExprField
- ir::ast::Field
- ir::ast::GraphBlock
- ir::ast::ImportStmt
- ir::ast::InferenceBlock
- ir::ast::InputDecl
- ir::ast::LoggingBlock
- ir::ast::MetadataBlock
- ir::ast::MetricDef
- ir::ast::MetricsBlock
- ir::ast::NodeDecl
- ir::ast::OutputDecl
- ir::ast::ParamAttr
- ir::ast::ParamDecl
- ir::ast::ParamDef
- ir::ast::PlotDef
- ir::ast::Program
- ir::ast::TrainingBlock
- ir::ast::TypeDef
- ir::ast::TypesBlock
- ir::ast::VisualizationBlock
- ir::error::Error
- ir::graph::GradClipConfig
- ir::graph::InferenceConfig
- ir::graph::IrAssert
- ir::graph::IrGraph
- ir::graph::IrNode
- ir::graph::IrOutput
- ir::graph::IrParam
- ir::graph::IrProgram
- ir::graph::LrScheduleConfig
- ir::graph::NodeId
- ir::graph::OptimizerConfig
- ir::graph::TrainingConfig
- ir::lexer::Lexer
- ir::optimize::OptStats
- ir::parser::Parser
- ir::token::Span
- ir::token::Token
- ir::validate::ValidationError
- nn::AdaptiveAvgPool2d
- nn::AvgPool2d
- nn::BatchNorm2d
- nn::ClassMetrics
- nn::ConfusionMatrix
- nn::Conv1d
- nn::Conv2d
- nn::Dropout
- nn::ELU
- nn::Embedding
- nn::Flatten
- nn::GRU
- nn::GRUCell
- nn::GeLU
- nn::GroupNorm
- nn::LSTM
- nn::LSTMCell
- nn::LayerNorm
- nn::LeakyReLU
- nn::Linear
- nn::MaxPool2d
- nn::Mish
- nn::MultiHeadAttention
- nn::RMSNorm
- nn::RNN
- nn::RNNCell
- nn::ReLU
- nn::Sequential
- nn::SiLU
- nn::Sigmoid
- nn::Tanh
- nn::TransformerBlock
- nn::activation::ELU
- nn::activation::GeLU
- nn::activation::LeakyReLU
- nn::activation::Mish
- nn::activation::ReLU
- nn::activation::SiLU
- nn::activation::Sigmoid
- nn::activation::Tanh
- nn::attention::MultiHeadAttention
- nn::batchnorm::BatchNorm2d
- nn::conv::AdaptiveAvgPool2d
- nn::conv::AvgPool2d
- nn::conv::Conv1d
- nn::conv::Conv2d
- nn::conv::MaxPool2d
- nn::dropout::Dropout
- nn::embedding::Embedding
- nn::flatten::Flatten
- nn::groupnorm::GroupNorm
- nn::layernorm::LayerNorm
- nn::linear::Linear
- nn::metrics::ClassMetrics
- nn::metrics::ConfusionMatrix
- nn::rmsnorm::RMSNorm
- nn::rnn::GRU
- nn::rnn::GRUCell
- nn::rnn::LSTM
- nn::rnn::LSTMCell
- nn::rnn::RNN
- nn::rnn::RNNCell
- nn::sequential::Sequential
- nn::transformer::TransformerBlock
- onnx::OnnxGraph
- onnx::OnnxModel
- onnx::OnnxNode
- onnx::OnnxTensor
- optim::Adam
- optim::AdamW
- optim::CosineAnnealingLR
- optim::CosineWarmupLR
- optim::EMA
- optim::ExponentialLR
- optim::GradAccumulator
- optim::LinearLR
- optim::OptimizerState
- optim::RAdam
- optim::RMSProp
- optim::ReduceLROnPlateau
- optim::SGD
- optim::StepLR
- optim::adam::Adam
- optim::adam::AdamW
- optim::clip::GradAccumulator
- optim::ema::EMA
- optim::optimizer::OptimizerState
- optim::radam::RAdam
- optim::rmsprop::RMSProp
- optim::scheduler::CosineAnnealingLR
- optim::scheduler::CosineWarmupLR
- optim::scheduler::ExponentialLR
- optim::scheduler::LinearLR
- optim::scheduler::ReduceLROnPlateau
- optim::scheduler::StepLR
- optim::sgd::SGD
- prelude::Adam
- prelude::AdamW
- prelude::AdaptiveAvgPool2d
- prelude::AvgPool2d
- prelude::BatchNorm2d
- prelude::Conv1d
- prelude::Conv2d
- prelude::CosineAnnealingLR
- prelude::CosineWarmupLR
- prelude::CpuBackend
- prelude::CpuDevice
- prelude::Dropout
- prelude::ELU
- prelude::EMA
- prelude::ExponentialLR
- prelude::Flatten
- prelude::GRU
- prelude::GRUCell
- prelude::GeLU
- prelude::GradAccumulator
- prelude::GradStore
- prelude::GroupNorm
- prelude::LSTM
- prelude::LSTMCell
- prelude::LayerNorm
- prelude::LeakyReLU
- prelude::Linear
- prelude::LinearLR
- prelude::MaxPool2d
- prelude::Mish
- prelude::MultiHeadAttention
- prelude::OptimizerState
- prelude::RAdam
- prelude::RMSNorm
- prelude::RMSProp
- prelude::RNN
- prelude::RNNCell
- prelude::ReLU
- prelude::ReduceLROnPlateau
- prelude::SGD
- prelude::Sequential
- prelude::Shape
- prelude::SiLU
- prelude::StepLR
- prelude::Tensor
- prelude::TransformerBlock
- profiler::BenchmarkResult
- profiler::MemoryTracker
- profiler::ModelSummary
- profiler::ProfileEntry
- profiler::ProfileEvent
- profiler::ProfileReport
- profiler::Profiler
- profiler::ScopedTimer
- profiler::Stopwatch
- quantize::QuantConfig
- quantize::QuantStats
- quantize::QuantizedLinear
- quantize::QuantizedTensor
Enums
- BinaryOp
- CmpOp
- CpuStorage
- CudaStorage
- DType
- Error
- Op
- ReduceOp
- UnaryOp
- distributed::AllReduceOp
- exec::jit::BinaryInstr
- exec::jit::CompareInstr
- exec::jit::Instruction
- exec::jit::LogicalBinInstr
- exec::jit::ReduceInstr
- exec::jit::UnaryInstr
- ir::ast::BinOp
- ir::ast::CustomOpStmt
- ir::ast::DTypeKind
- ir::ast::Dimension
- ir::ast::Expr
- ir::ast::GraphStmt
- ir::ast::HintKind
- ir::ast::InferenceField
- ir::ast::Literal
- ir::ast::NodeStmt
- ir::ast::TopLevel
- ir::ast::TrainingField
- ir::ast::TypeExpr
- ir::ast::UnaryOp
- ir::error::ErrorKind
- ir::graph::AttrValue
- ir::graph::ConfigValue
- ir::graph::ConstantValue
- ir::graph::DType
- ir::graph::Dim
- ir::graph::InitStrategy
- ir::graph::IrHint
- ir::graph::IrType
- ir::graph::OpKind
- ir::token::TokenKind
- ir::validate::ValidationErrorKind
- nn::Average
- nn::Reduction
- nn::init::FanMode
- nn::loss::Reduction
- nn::metrics::Average
- onnx::OnnxAttribute
- prelude::DType
- quantize::QuantBits
- quantize::QuantGranularity
- quantize::QuantMode
Traits
- Backend
- BackendDevice
- BackendStorage
- WithDType
- nn::Module
- nn::module::Module
- optim::LrScheduler
- optim::Optimizer
- optim::Stateful
- optim::optimizer::Optimizer
- optim::optimizer::Stateful
- optim::scheduler::LrScheduler
- prelude::LrScheduler
- prelude::Module
- prelude::Optimizer
- prelude::Stateful
Functions
- checkpoint::from_bytes
- checkpoint::load
- checkpoint::load_tensors
- checkpoint::load_trainer
- checkpoint::load_training
- checkpoint::read_checkpoint
- checkpoint::read_training_checkpoint
- checkpoint::save
- checkpoint::save_tensors
- checkpoint::save_trainer
- checkpoint::save_training
- checkpoint::to_bytes
- checkpoint::training_from_bytes
- checkpoint::training_to_bytes
- checkpoint::write_checkpoint
- checkpoint::write_training_checkpoint
- distributed::reduce_gradients
- exec::jit::compile_graph
- exec::jit::load_jit
- exec::load_program
- exec::load_trainer
- ir::infer_shapes
- ir::lower
- ir::lower::lower
- ir::optimize
- ir::optimize::eliminate_common_subexprs
- ir::optimize::eliminate_dead_code
- ir::optimize::eliminate_identities
- ir::optimize::fold_constants
- ir::optimize::fuse_operators
- ir::optimize::optimize
- ir::optimize::optimize_graph
- ir::optimize::optimize_graph_with_stats
- ir::parse
- ir::parser::parse
- ir::shapes::infer_graph_shapes
- ir::shapes::infer_shapes
- ir::token::keyword_lookup
- ir::validate
- ir::validate::validate
- ir::validate::validate_graph_standalone
- nn::accuracy
- nn::argmax_classes
- nn::bce_loss
- nn::bce_with_logits_loss
- nn::classification_report
- nn::cross_entropy_loss
- nn::f1_score
- nn::init::constant
- nn::init::kaiming_normal
- nn::init::kaiming_uniform
- nn::init::normal
- nn::init::ones
- nn::init::uniform
- nn::init::xavier_normal
- nn::init::xavier_uniform
- nn::init::zeros
- nn::l1_loss
- nn::l1_loss_with_reduction
- nn::loss::bce_loss
- nn::loss::bce_with_logits_loss
- nn::loss::cross_entropy_loss
- nn::loss::l1_loss
- nn::loss::l1_loss_with_reduction
- nn::loss::mse_loss
- nn::loss::mse_loss_with_reduction
- nn::loss::nll_loss
- nn::loss::smooth_l1_loss
- nn::mae
- nn::mape
- nn::metrics::accuracy
- nn::metrics::argmax_classes
- nn::metrics::classification_report
- nn::metrics::f1_score
- nn::metrics::mae
- nn::metrics::mape
- nn::metrics::perplexity
- nn::metrics::perplexity_from_log_probs
- nn::metrics::precision
- nn::metrics::r2_score
- nn::metrics::recall
- nn::metrics::rmse
- nn::metrics::tensor_accuracy
- nn::metrics::top_k_accuracy
- nn::mse_loss
- nn::mse_loss_with_reduction
- nn::nll_loss
- nn::perplexity
- nn::perplexity_from_log_probs
- nn::precision
- nn::r2_score
- nn::recall
- nn::rmse
- nn::smooth_l1_loss
- nn::tensor_accuracy
- nn::top_k_accuracy
- onnx::export_tensors
- onnx::export_weights
- onnx::load_onnx_graph
- onnx::load_onnx_graph_from_bytes
- onnx::load_onnx_weights
- onnx::load_onnx_weights_from_bytes
- onnx::run_onnx_graph
- optim::clip::clip_grad_norm
- optim::clip::clip_grad_value
- optim::clip::grad_norm
- optim::clip_grad_norm
- optim::clip_grad_value
- optim::grad_norm
- prelude::bce_loss
- prelude::bce_with_logits_loss
- prelude::clip_grad_norm
- prelude::clip_grad_value
- prelude::cross_entropy_loss
- prelude::grad_norm
- prelude::l1_loss
- prelude::mse_loss
- prelude::nll_loss
- prelude::smooth_l1_loss
- profiler::benchmark_forward
- profiler::benchmark_forward_backward
- profiler::estimate_model_memory
- profiler::format_bytes
- quantize::dequantize_tensor
- quantize::quantization_stats
- quantize::quantize_named_parameters
- quantize::quantize_tensor
- safetensors::from_bytes
- safetensors::load
- safetensors::load_state_dict
- safetensors::read_safetensors
- safetensors::save
- safetensors::save_module
- safetensors::to_bytes
- safetensors::write_safetensors