tiny-recursive-rs 0.1.0

Rust implementation of Tiny Recursive Models for efficient puzzle solving
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13

/// Training infrastructure for TRM

// Note: Using candle-nn's built-in AdamW optimizer instead of custom implementation
// pub mod optimizer;
pub mod scheduler;
pub mod ema;
pub mod checkpoint;
pub mod trainer;

pub use scheduler::CosineScheduler;
pub use ema::EMA;
pub use checkpoint::Checkpoint;
pub use trainer::{Trainer, TrainingConfig};