candle-nn/src/lib.rs


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18

// For now this crate shares its error type with candle-core. We may introduce some separate
// error type if needed or add some specialized cases on the candle-core side.
pub mod activation;
pub mod conv;
pub mod embedding;
pub mod init;
pub mod layer_norm;
pub mod linear;
pub mod optim;
pub mod var_builder;

pub use activation::Activation;
pub use conv::{Conv1d, Conv1dConfig};
pub use embedding::Embedding;
pub use layer_norm::LayerNorm;
pub use linear::Linear;
pub use optim::SGD;
pub use var_builder::VarBuilder;