List of all items

Structs

• Frozen
• Parameters
• layers::Embedding
• layers::LayerNorm
• layers::Linear
• layers::LoraLinear
• layers::QuantizedEmbedding
• layers::QuantizedLinear
• layers::RmsNorm
• layers::Rope
• optim::Adam
• optim::Lion
• optim::Muon
• optim::Sgd

Enums

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

Traits

• Module
• optim::Optimizer

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
• value_and_grad

Constants

• VERSION