Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation
Python
82
389 commits
updated Feb 12, 2026
Contributors
Languages
Python
62.1%
TeX
33.1%
C++
4.2%
Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation
Python
82
389 commits
updated Feb 12, 2026
Contributors
Languages
Python
62.1%
TeX
33.1%
C++
4.2%