bitnet_quantize/layer/mod.rs
1//! Neural network layers for BitNet.
2//!
3//! This module provides:
4//! - `BitLinear`: Drop-in replacement for `nn::Linear` with ternary weights
5//! - Straight-Through Estimator for training
6
7mod bitlinear;
8mod ste;
9
10pub use bitlinear::BitLinear;
11pub use ste::{int8_ste, ste_backward, ste_forward, ternary_ste};