tests/jax/test_distributed_layernorm_mlp.py · 44740c6cc8716ac6fb141775a4ab478e9a4143ae · OpenDAS / TransformerEngine · GitLab

Find file Blame History Permalink

[JAX] Update tolerance of distributed layernorm MLP for FP8 (#1971) · ca7407e3
jberchtold-nvidia authored Jul 18, 2025
```
Update tolerance of distributed layernorm MLP for FP8
Signed-off-by: Jeremy Berchtold <jberchtold@nvidia.com>
```
ca7407e3

test_distributed_layernorm_mlp.py 18.8 KB