Skip to content

Add support for JAX FFI vmap in jax_kernel() and jax_callable() (GH-859)#1203

Closed
nvlukasz wants to merge 5 commits intoNVIDIA:mainfrom
nvlukasz:lwawrzyniak/jax-ffi-vmap
Closed

Add support for JAX FFI vmap in jax_kernel() and jax_callable() (GH-859)#1203
nvlukasz wants to merge 5 commits intoNVIDIA:mainfrom
nvlukasz:lwawrzyniak/jax-ffi-vmap

Commits

Commits on Jan 30, 2026