JacQuant: STE-Free Quantization-Aware Training via Learned Jacobian Surrogates

Kai Yi, Vignesh Vivekraja, Harshit Khaitan +1

Quantization-aware training (QAT) is widely deployed but typically relies on the Straight-Through Estimator (STE), which passes gradients through non-differentiable quantizers by f…