List of all items
Structs
- AdaGrad
- AdaGradConfig
- AdaGradState
- Adam
- AdamConfig
- AdamState
- AdamW
- AdamWConfig
- AdamWState
- Adan
- AdanConfig
- AdanState
- AdaptiveMomentumState
- AdaptiveNesterovMomentumState
- CenteredState
- DynState
- GradientsAccumulator
- GradientsParams
- LBFGS
- LBFGSConfig
- LBFGSState
- LrDecayState
- ModuleOptimizer
- MultiGradientsParams
- Muon
- MuonConfig
- MuonState
- OptimizerRecord
- RmsProp
- RmsPropConfig
- RmsPropMomentum
- RmsPropMomentumState
- RmsPropState
- Sgd
- SgdConfig
- SgdState
- SquareAvgState
- StateSink
- StateSource
- decay::WeightDecay
- decay::WeightDecayConfig
- decay::WeightDecayState
- lr_scheduler::DynLrScheduler
- lr_scheduler::LrSchedulerRecord
- lr_scheduler::composed::ComposedLrScheduler
- lr_scheduler::composed::ComposedLrSchedulerConfig
- lr_scheduler::constant::ConstantLr
- lr_scheduler::cosine::CosineAnnealingLrScheduler
- lr_scheduler::cosine::CosineAnnealingLrSchedulerConfig
- lr_scheduler::cosine::CosineAnnealingLrSchedulerState
- lr_scheduler::exponential::ExponentialLrScheduler
- lr_scheduler::exponential::ExponentialLrSchedulerConfig
- lr_scheduler::exponential::ExponentialLrSchedulerState
- lr_scheduler::linear::LinearLrScheduler
- lr_scheduler::linear::LinearLrSchedulerConfig
- lr_scheduler::linear::LinearLrSchedulerState
- lr_scheduler::module_lr_scheduler::ModuleLearningRate
- lr_scheduler::module_lr_scheduler::ModuleLrScheduler
- lr_scheduler::module_lr_scheduler::ModuleLrSchedulerConfig
- lr_scheduler::noam::NoamLrScheduler
- lr_scheduler::noam::NoamLrSchedulerConfig
- lr_scheduler::noam::NoamLrSchedulerState
- lr_scheduler::sequential::SequentialLrScheduler
- lr_scheduler::sequential::SequentialLrSchedulerConfig
- lr_scheduler::step::StepLrScheduler
- lr_scheduler::step::StepLrSchedulerConfig
- lr_scheduler::step::StepLrSchedulerState
- module_optimizer::ModuleOptimizer
- momentum::Momentum
- momentum::MomentumConfig
- momentum::MomentumState
- record::OptimizerRecord
Enums
- AdjustLrFn
- GradAdaptor
- LineSearchFn
- grad_clipping::GradientClipping
- grad_clipping::GradientClippingConfig
- lr_scheduler::LrSchedulerConfig
- lr_scheduler::composed::SchedulerReduction
- module_optimizer::GradAdaptor