Commits · 6182206102a553ca73aaa13b499d0fb3f40fdd73 · OpenDAS / TransformerEngine

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