build_tools/jax.py · d2973cb8b49ad0dd15ebbdfba0ec9e9c23cee818 · OpenDAS / TransformerEngine

Kirthi Shankar Sivamani authored May 14, 2025


Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
Co-authored-by: jberchtold-nvidia <158520091+jberchtold-nvidia@users.noreply.github.com>

d2973cb8

jax.py 2.4 KB

Replace jax.py