transformer_engine/jax/quantize/scaling_modes.py · 063ef88d1af050b9076daaa5b92ed2d0018796fa · OpenDAS / TransformerEngine

jberchtold-nvidia authored Oct 09, 2025


Signed-off-by: Jeremy Berchtold <jberchtold@nvidia.com>
Co-authored-by: Phuong Nguyen <phuonguyen@nvidia.com>

8a7ab3dd

scaling_modes.py 38.7 KB