bitnet-quantize 0.2.1

Microsoft BitNet b1.58 quantization and inference for Rust
Documentation 
1
2
3
4
5
6
7
8
9
10
11

//! Neural network layers for BitNet.
//!
//! This module provides:
//! - `BitLinear`: Drop-in replacement for `nn::Linear` with ternary weights
//! - Straight-Through Estimator for training

mod bitlinear;
mod ste;

pub use bitlinear::BitLinear;
pub use ste::{int8_ste, ste_backward, ste_forward, ternary_ste};