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;