examples/jax/encoder/test_model_parallel_encoder.py · 2236292a4cf2e6f276ba8444bac1125a3313f394 · OpenDAS / TransformerEngine · GitLab

Find file Blame History Permalink

[JAX] Disable fused attention in encoder tests for determinism (#2601) · 2236292a
jberchtold-nvidia authored Jan 15, 2026
```
disable fused attention in encoder tests for determinism
Signed-off-by: Jeremy Berchtold <jberchtold@nvidia.com>
```
2236292a

test_model_parallel_encoder.py 23.2 KB