smelte_rs/nn/layers/
mod.rs

1/// Linear layers
2pub mod linear;
3
4/// Layer norm
5pub mod layer_norm;
6
7/// Embedding
8pub mod embedding;
9
10pub use embedding::Embedding;
11pub use layer_norm::LayerNorm;
12pub use linear::{Linear, LinearT, UnbiasedLinear};