mirror of
https://github.com/ROCm/jax.git
synced 2025-04-24 20:26:05 +00:00
dda6430f7c
This function is (for better or worse) user facing for custom call users. I think it's worth having this in the API docs.
27 lines
488 B
ReStructuredText
27 lines
488 B
ReStructuredText
``jax.lib`` module
|
|
==================
|
|
The `jax.lib` package is a set of internal tools and types for bridging between
|
|
JAX's Python frontend and its XLA backend.
|
|
|
|
jax.lib.xla_bridge
|
|
------------------
|
|
|
|
.. currentmodule:: jax.lib.xla_bridge
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
default_backend
|
|
get_backend
|
|
get_compile_options
|
|
|
|
jax.lib.xla_client
|
|
------------------
|
|
|
|
.. currentmodule:: jaxlib.xla_client
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
register_custom_call_target
|