Module training 

Re-exports

pub use scheduler::CosineScheduler;

pub use ema::EMA;

pub use checkpoint::Checkpoint;

pub use trainer::Trainer;

pub use trainer::TrainingConfig;

Modules

checkpoint

ema

scheduler

Training infrastructure for TRM

trainer