qa/L2_jax_distributed_unittest/test.sh · c1a1c04e6341778562a45cff847dcaface8b33cc · OpenDAS / TransformerEngine · GitLab

Find file Blame History Permalink

[JAX] Ensure JAX reference impl uses an accurate backend in our tests (#2322) · 1269b2e2
jberchtold-nvidia authored Oct 30, 2025
```
Ensure JAX reference impl uses an accurate backend
Signed-off-by: Jeremy Berchtold <jberchtold@nvidia.com>
```
1269b2e2

test.sh 542 Bytes