rust_trainer 0.1.4

CPU-first pure-Rust supervised trainer for Selective State Space Models with Hyperspherical Prototype Networks.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23

pub mod data_stream;
pub mod generic_trainer;
pub mod layer;
pub mod loss;
pub mod nn;
pub mod optim;
pub mod simd_ops;
pub mod stack;
pub mod trainer;

#[cfg(feature = "python")]
pub mod bindings;

pub use loss::{
    cagradstep, gradnorm_ff_scale, pcgrad, GradientSurgeryConfig, GradientSurgeryMethod,
};
pub use trainer::{
    AdamWConfig, CadencedStepStats, ExpansionConfig, ExpansionPlacement, ExperimentalTrainer,
    ExperimentalTrainerConfig, FreezeSelection, LayerSpec, MambaLayerParams, TrainerParams,
};

#[cfg(feature = "python")]
pub use bindings::{train_from_config, TrainerConfig};