2024-12-20 11:26:04 +00:00
|
|
|
``jax.ffi`` module
|
|
|
|
==================
|
|
|
|
|
|
|
|
.. automodule:: jax.ffi
|
|
|
|
|
|
|
|
.. autosummary::
|
|
|
|
:toctree: _autosummary
|
|
|
|
|
|
|
|
ffi_call
|
|
|
|
ffi_lowering
|
|
|
|
pycapsule
|
|
|
|
register_ffi_target
|
2025-01-07 07:28:21 -08:00
|
|
|
register_ffi_type_id
|
2024-12-20 11:26:04 +00:00
|
|
|
|
|
|
|
|
|
|
|
``jax.extend.ffi`` module (deprecated)
|
|
|
|
======================================
|
|
|
|
|
|
|
|
The ``jax.extend.ffi`` module has been moved to ``jax.ffi``, and that import
|
|
|
|
path should be used instead, but these functions remain documented here while
|
|
|
|
the legacy import is being deprecated.
|
|
|
|
|
|
|
|
.. automodule:: jax.extend.ffi
|
|
|
|
|
|
|
|
.. autosummary::
|
|
|
|
:toctree: _autosummary
|
|
|
|
|
|
|
|
ffi_call
|
|
|
|
ffi_lowering
|
|
|
|
pycapsule
|
|
|
|
register_ffi_target
|