transformer_engine/jax/cpp_extensions/quantization.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

quantization.py 43 KB