List of all items

Structs

• OneHotEncoder
• layer_spec::Conv1d
• layer_spec::Cosine
• layer_spec::Dense
• layer_spec::DyT
• layer_spec::GLU
• layer_spec::GLULeakyRelu
• layer_spec::LRDiv
• layer_spec::LeakyRelu
• layer_spec::Linear
• layer_spec::RMSNorm
• layer_spec::Relu
• layer_spec::SiLU
• layer_spec::Sigmoid
• layer_spec::Sine
• layer_spec::Softmax
• layer_spec::Softplus
• layer_spec::SwiGLU
• layer_spec::Swish
• layer_spec::Tanh
• layer_spec::TanhGLU
• prelude::TrainParams
• problem::AxPlusB
• problem::ModularAddition10
• problem::ModularAddition16
• problem::ModularAddition32
• problem::Parity
• problem::idx::Reader
• problem::idx::RowIterator
• problem::mnist::ImgClassification
• recorder::BatchInfo
• recorder::Recorder
• recorder::RecorderBuilder

Enums

• problem::idx::Element
• recorder::Every
• recorder::Record

Traits

• Buildable
• prelude::BackpropModule
• prelude::LoadableModule
• prelude::Module
• prelude::ResetParams
• prelude::TracedModule
• prelude::loss::CosineLoss
• prelude::loss::DiffLoss
• prelude::loss::LogitLoss
• problem::Problem

Functions

• prelude::train_batch
• prelude::train_batch_parallel
• prelude::train_step
• train_batch
• train_batch_parallel
• train_step

Type Aliases

• prelude::Error