mirror of
https://github.com/ROCm/jax.git
synced 2025-04-18 12:56:07 +00:00
b308c64936
This means that users of the FFI interface won't need to directly interact with `jaxlib.xla_client` at all. I've expanded the doctring a little and changed one default: the default `api_version` is `1` instead of `0` to be consistent with the new name.
13 lines
188 B
ReStructuredText
13 lines
188 B
ReStructuredText
``jax.extend.ffi`` module
|
|
=========================
|
|
|
|
.. automodule:: jax.extend.ffi
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
ffi_call
|
|
ffi_lowering
|
|
pycapsule
|
|
register_ffi_target
|