candle-nn/src/lib.rs


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

// 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.
mod activation;
mod conv;
mod embedding;
mod layer_norm;
mod linear;
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 var_builder::VarBuilder;