Re-exports§
pub use loss::cagradstep;pub use loss::gradnorm_ff_scale;pub use loss::pcgrad;pub use loss::GradientSurgeryConfig;pub use loss::GradientSurgeryMethod;pub use trainer::AdamWConfig;pub use trainer::CadencedStepStats;pub use trainer::ExpansionConfig;pub use trainer::ExpansionPlacement;pub use trainer::ExperimentalTrainer;pub use trainer::ExperimentalTrainerConfig;pub use trainer::FreezeSelection;pub use trainer::LayerSpec;pub use trainer::MambaLayerParams;pub use trainer::TrainerParams;