examples/jax/encoder/test_multigpu_encoder.py · 44740c6cc8716ac6fb141775a4ab478e9a4143ae · OpenDAS / TransformerEngine · GitLab

Find file Blame History Permalink

[JAX] Tighten Encoder Test tolerances (#1955) · ed75c2b0
Phuong Nguyen authored Jul 17, 2025
```
tighten encoder test tols
Signed-off-by: Phuong Nguyen <phuonguyen@nvidia.com>
```
ed75c2b0

test_multigpu_encoder.py 18.3 KB