vyre-libs 0.7.2

vyre Category A library ecosystem - pure-IR compositions over foundation IR and primitive-owned kernels
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14

//! Normalization sub-dialect: LayerNorm, RMSNorm, layerwise LN scale.
mod gated_rms_norm;
mod last_dim_l2_norm;
mod layer_norm;
pub mod layerwise_ln_scale;
mod rms_norm;

pub use gated_rms_norm::{
    gated_rms_norm, gated_rms_norm_with_weight_dtype, learned_rms_norm, GatedRmsNormError,
};
pub use last_dim_l2_norm::{last_dim_l2_norm, LastDimL2NormError};
pub use layer_norm::{layer_norm, LayerNorm};
pub use layerwise_ln_scale::layerwise_ln_scale;
pub use rms_norm::{rms_norm, rms_norm_reference};