jax.extend.backend module

.. automodule:: jax.extend.backend

.. autosummary:: :toctree: _autosummary

backends backend_xla_version clear_backends get_backend get_compile_options get_default_device ifrt_proxy register_backend_cache register_backend_factory