Skip to main content

Module model

Module model 

Source
Expand description

The model surface: write a network, train it, checkpoint it.

Everything here re-exports the crate root — use topos::model::* is enough to record, run, and train, and flat use topos::Tape imports keep working unchanged.

Re-exports§

pub use crate::Activation;
pub use crate::Adam;
pub use crate::AdamW;
pub use crate::Adjoints;
pub use crate::BatchNorm;
pub use crate::BoundEntry;
pub use crate::Conv2d;
pub use crate::Detach;
pub use crate::Dropout;
pub use crate::Embedding;
pub use crate::Entry;
pub use crate::Field;
pub use crate::Gradients;
pub use crate::LayerNorm;
pub use crate::Linear;
pub use crate::Mlp;
pub use crate::Module;
pub use crate::Network;
pub use crate::Normalization;
pub use crate::Optimizer;
pub use crate::Parameters;
pub use crate::Path;
pub use crate::Plan;
pub use crate::RmsNorm;
pub use crate::Run;
pub use crate::Segment;
pub use crate::Sequential;
pub use crate::Sgd;
pub use crate::Shape;
pub use crate::Symbol;
pub use crate::Tape;
pub use crate::Tensor;
pub use crate::Value;
pub use crate::Visitor;
pub use crate::causal_mask;
pub use crate::checkpoint;
pub use crate::concat;
pub use crate::conv2d;
pub use crate::cross_entropy;
pub use crate::init;
pub use crate::max_pool;
pub use crate::named_parameters;
pub use crate::parameters;
pub use crate::scaled_dot_product;
pub use crate::stack;