topos 0.12.0

An autodiff compiler stack in Rust.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34

mod activation;
mod adam;
mod batch_norm;
// Public modules where the names are meaningless unqualified:
// initializer names (`uniform`, `normal`) need `init::`, and
// checkpoint verbs (`snapshot`, `restore`) need `checkpoint::`.
pub mod checkpoint;
mod convolution;
mod dropout;
pub mod init;
mod layer_norm;
mod linear;
mod loss;
mod mlp;
mod module;
mod optimizer;
mod pooling;
mod rms_norm;
mod sequential;

pub use activation::Activation;
pub use adam::{Adam, AdamW};
pub use batch_norm::{BatchNorm, Normalization};
pub use convolution::{Conv2d, conv2d};
pub use dropout::Dropout;
pub use layer_norm::LayerNorm;
pub use linear::Linear;
pub use loss::cross_entropy;
pub use mlp::Mlp;
pub use module::{Module, Path, Segment, Visitor, named_parameters, parameters};
pub use optimizer::{Optimizer, Sgd};
pub use pooling::max_pool;
pub use rms_norm::RmsNorm;
pub use sequential::Sequential;