smelte-rs 0.1.0

Efficient inference ML framework written in rust
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12

/// Linear layers
pub mod linear;

/// Layer norm
pub mod layer_norm;

/// Embedding
pub mod embedding;

pub use embedding::Embedding;
pub use layer_norm::LayerNorm;
pub use linear::{Linear, LinearT, UnbiasedLinear};