mirror of
https://github.com/ROCm/jax.git
synced 2025-04-16 20:06:05 +00:00
a7f384cc6e
This enables dynamic registration of custom FFI types on the appropriate platform via PJRT. PiperOrigin-RevId: 712904085
32 lines
601 B
ReStructuredText
32 lines
601 B
ReStructuredText
``jax.ffi`` module
|
|
==================
|
|
|
|
.. automodule:: jax.ffi
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
ffi_call
|
|
ffi_lowering
|
|
pycapsule
|
|
register_ffi_target
|
|
register_ffi_type_id
|
|
|
|
|
|
``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
|