jax
fa46bbf0 - Add pycapsule utility to jax.extend.ffi

Commit
2 years ago
Add pycapsule utility to jax.extend.ffi This allows one to build PyCapsules ready for XLA custom call registration out of function pointers retrieved from shared libraries using ctypes. In particular, this obviates the need to create Python bindings to custom call targets.
Author
Committer
Parents
Loading