https://github.com/google/jax
Raw File
Tip revision: fed7efd73003988282744b2f649df493b808c781 authored by jax authors on 03 April 2024, 18:45:56 UTC
Merge pull request #20571 from hawkinsp:release
Tip revision: fed7efd
jax.lib.rst
``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
back to top