tests/jax/test_distributed_layernorm.py · 05d3b7b511ff142a2a0e6d7b46d56b81d1323461 · OpenDAS / TransformerEngine

[JAX] Remove unneccessary MXFP8 scale_inv padding (#1954) · 5350f277

Phuong Nguyen authored Jul 17, 2025



* remove unnecessary padding
Signed-off-by: Phuong Nguyen <phuonguyen@nvidia.com>

* adapt the test_distributed_layernorm byte count
Signed-off-by: Phuong Nguyen <phuonguyen@nvidia.com>


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

5350f277

test_distributed_layernorm.py 10.1 KB

Replace test_distributed_layernorm.py