tests/jax/test_distributed_layernorm_mlp.py · 11fecc4140010a4bba8fe3bfe06b9a3f4449b9d9 · OpenDAS / TransformerEngine · GitLab

Find file Blame History Permalink

[JAX] Update distributed LayerNormMLP test tolerance for L40 (#1901) · 11fecc41
jberchtold-nvidia authored Jul 11, 2025
```
Update test tolerance for L40
Signed-off-by: Jeremy Berchtold <jberchtold@nvidia.com>
```
11fecc41

test_distributed_layernorm_mlp.py 14.6 KB