tests/jax/test_distributed_layernorm_mlp.py · ca7407e3e78bb692ffbec552d1d71e2fd8901259 · OpenDAS / TransformerEngine · GitLab

"transformer_engine/jax/layernorm_mlp.py" did not exist on "eed4dfc67878d878af85f2d839d4db6aef3521a2"

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