Skip to main content

entrenar/quant/fake_quantize/
mod.rs

1//! Fake Quantization for Quantization-Aware Training (QAT)
2//!
3//! Fake quantization simulates the effects of quantization during training:
4//! - Forward: quantize → dequantize (simulates quantization noise)
5//! - Backward: Straight-Through Estimator (STE) passes gradients unchanged
6//!
7//! This allows models to adapt to quantization noise during training,
8//! resulting in better accuracy after actual quantization.
9
10mod config;
11mod ops;
12mod quantize;
13
14#[cfg(test)]
15mod tests;
16
17// Re-export all public types
18pub use config::FakeQuantConfig;
19pub use ops::{fake_quantize, ste_backward};
20pub use quantize::FakeQuantize;