NIFTy-PPL/JAXbind

Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation