transformer_engine/jax/cpp_extensions/misc.py · 2643ba1df43397cc84c9da5fe719a66d87ad9a0a · OpenDAS / TransformerEngine · GitLab

Find file Blame History Permalink

[TE/JAX] Update required JAX version for FFI custom calls with cudaGraph (#1285) · 7cef7566
Phuong Nguyen authored Oct 25, 2024
```
Update jax version for ffi
Signed-off-by: Phuong Nguyen <phuonguyen@nvidia.com>
```
7cef7566

misc.py 5.25 KB