rwkvzip/rwkv7/training/
mod.rs

1//! RWKV7 training (byte-level) implemented in Rust.
2//!
3//! This module trains a RWKV7 model that is *weight-compatible* with the
4//! SIMD inference path in `crate::rwkv7::Model` by exporting the same
5//! safetensors keys & shapes.
6//!
7//! Design goals:
8//! - Correctness first: math matches `src/rwkv7/model.rs`
9//! - GPU acceleration via libtorch (tch) when available
10//! - Small-model friendliness (<10M params) with stable defaults
11
12mod data;
13mod export;
14mod model;
15mod model_fast;
16mod train;
17mod validate;
18mod wkv_cuda;
19mod wkv_fused;
20
21pub use train::{train_enwik8, TrainConfig, TrainReport};
22pub use wkv_cuda::cuda_wkv_available;