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