List of all items

Structs

• ModuleDisplay
• Param
• Summary
• SummaryRow
• layers::Dropout
• layers::Embedding
• layers::LayerNorm
• layers::Linear
• layers::LoraLinear
• layers::QuantizedEmbedding
• layers::QuantizedLinear
• layers::RmsNorm
• layers::Rope
• optim::Adam
• optim::AdamSlot
• optim::AdamW
• optim::Lion
• optim::MomentumSlot
• optim::MultiOptimizer
• optim::Muon
• optim::OptimizerState
• optim::Sgd
• optim::SgdSlot

Enums

• layers::LoraBase
• losses::Reduction
• optim::LearningRate

Traits

• Module
• Tree
• WeightSource
• optim::Optimizer

Derive Macros

• Module
• Tree

Functions

• gelu_approx
• gelu_fast_approx
• losses::cross_entropy
• optim::clip_grad_norm
• optim::schedules::cosine_decay
• optim::schedules::exponential_decay
• optim::schedules::join_schedules
• optim::schedules::linear_schedule
• optim::schedules::step_decay
• relu
• silu

Constants

• VERSION