tiny_recursive_rs/training/
mod.rs

1/// Training infrastructure for TRM
2
3// Note: Using candle-nn's built-in AdamW optimizer instead of custom implementation
4// pub mod optimizer;
5pub mod scheduler;
6pub mod ema;
7pub mod checkpoint;
8pub mod trainer;
9
10pub use scheduler::CosineScheduler;
11pub use ema::EMA;
12pub use checkpoint::Checkpoint;
13pub use trainer::{Trainer, TrainingConfig};