transformer_engine/jax/cpp_extensions/quantization.py · ac4e0fd63afb1998904695e1321c5631192c3a85 · OpenDAS / TransformerEngine

Phuong Nguyen authored Oct 01, 2025



* rm using_global_amax_of_x
Signed-off-by: Phuong Nguyen <phuonguyen@nvidia.com>

---------
Signed-off-by: Phuong Nguyen <phuonguyen@nvidia.com>

ac4e0fd6

quantization.py 38.8 KB