Commits · eb8e792b38228e3c221a2d6b69babe3a91acd1f4 · OpenDAS / TransformerEngine

05 Dec, 2025 1 commit
- [JAX] Enable TE/JAX test timings in CI (#2475) · 50be0299
  jberchtold-nvidia authored Dec 04, 2025
```
enable jax test timings in qa scripts
Signed-off-by: Jeremy Berchtold <jberchtold@nvidia.com>
```
  50be0299
30 Oct, 2025 1 commit
- [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
13 Aug, 2025 1 commit

[JAX] Add L2_jax_distributed_unittest (#2060) · ec65ba3c

jberchtold-nvidia authored Aug 12, 2025



* Add L2_jax_distributed_unittest
Signed-off-by: Jeremy Berchtold <jberchtold@nvidia.com>

* Add L1 entry for NORM_INPUT_SHAPES that was missing
Signed-off-by: Jeremy Berchtold <jberchtold@nvidia.com>

---------
Signed-off-by: Jeremy Berchtold <jberchtold@nvidia.com>

ec65ba3c