warp.jax_experimental.ffi
=========================

.. automodule:: warp.jax_experimental.ffi
   :no-members:

.. currentmodule:: warp.jax_experimental.ffi

API
---

.. autosummary::
   :nosignatures:
   :toctree: _generated

   GraphMode
   clear_jax_callable_graph_cache
   get_jax_callable_default_graph_cache_max
   jax_callable
   jax_kernel
   register_ffi_callback
   set_jax_callable_default_graph_cache_max