Skip to content

Fix compatibility with flax-basic_example.ipynb after JAX update#6247

Merged
JanuszL merged 3 commits intoNVIDIA:mainfrom
JanuszL:fix_dlpack_jax
Mar 10, 2026
Merged

Fix compatibility with flax-basic_example.ipynb after JAX update#6247
JanuszL merged 3 commits intoNVIDIA:mainfrom
JanuszL:fix_dlpack_jax

Commits

Commits on Mar 9, 2026

Commits on Mar 10, 2026