transformer_engine/jax/quantize/helper.py · e99be1b6af1aa138194c211ad9952858b3aaee44 · OpenDAS / TransformerEngine

[JAX] Fix code block in fp8_autocast docstring (#2228) · f936c2ac

jberchtold-nvidia authored Oct 02, 2025



Fix code block in fp8_autocast docstring
Signed-off-by: Jeremy Berchtold <jberchtold@nvidia.com>
Co-authored-by: Phuong Nguyen <phuonguyen@nvidia.com>

f936c2ac

helper.py 21.8 KB

Replace helper.py