build_tools/jax.py · 9d031fbd9dd8f9e57d8145080943d512c8b76840 · OpenDAS / TransformerEngine

Phuong Nguyen authored Jul 08, 2025



* Fix jax build
Signed-off-by: Phuong Nguyen <phuonguyen@nvidia.com>

---------
Signed-off-by: Phuong Nguyen <phuonguyen@nvidia.com>

9d031fbd

jax.py 2.39 KB