| """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"] |