List of all items

Structs

• config::TRMConfig
• data::numpy_dataset::DatasetMetadata
• data::numpy_dataset::NumpyDataLoader
• data::numpy_dataset::NumpyDataset
• layers::activations::CastedLinear
• layers::activations::LinearSwish
• layers::activations::SwiGLU
• layers::attention::Attention
• layers::embeddings::CastedEmbedding
• layers::normalization::RMSNorm
• layers::positional::RotaryEmbedding
• models::InnerCarry
• models::ReasoningModule
• models::TinyRecursiveModel
• models::TransformerBlock
• training::checkpoint::Checkpoint
• training::checkpoint::CheckpointMetadata
• training::ema::EMA
• training::ema::EMAConfig
• training::scheduler::CosineScheduler
• training::scheduler::CosineSchedulerConfig
• training::trainer::Trainer
• training::trainer::TrainingConfig

Enums

• TRMError

Traits

• data::BatchDataLoader

Functions

• layers::normalization::rms_norm
• layers::positional::apply_rotary_pos_emb
• models::loader::load_model
• training::checkpoint::load_checkpoint
• training::checkpoint::save_checkpoint
• utils::count_parameters
• utils::create_causal_mask
• utils::trunc_normal_init

Type Aliases

• Result