Add support for JAX FFI vmap in jax_kernel() and jax_callable() (GH-859)#1203
Closed
nvlukasz wants to merge 5 commits intoNVIDIA:mainfrom
Closed
Add support for JAX FFI vmap in jax_kernel() and jax_callable() (GH-859)#1203nvlukasz wants to merge 5 commits intoNVIDIA:mainfrom
nvlukasz wants to merge 5 commits intoNVIDIA:mainfrom
Commits
Commits on Jan 30, 2026
- committed
- committed
- committed
- committed
- committed