File size: 321 Bytes
e9c8366 | 1 2 3 4 5 6 7 8 9 | """Training utilities — STE (Straight-Through Estimator).
QAT for the unified architecture lives in training_unified.py
(UnifiedQATWrapper). This subpackage only exports the low-level STE primitive.
"""
from agiws_neural_quant.training.ste import fake_quantize, STEQuantize
__all__ = ["fake_quantize", "STEQuantize"] |