examples/jax/encoder/test_model_parallel_encoder.py · d972e76d0688b7f3441df15fc6e36555106c4817 · 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_model_parallel_encoder.py 20.8 KB