``jax.extend`` module

docs/jax.extend.rst

0.3.25251 B

.. currentmodule:: jax.extend

`jax.extend` module

.. automodule:: jax.extend

.. toctree:: :maxdepth: 1

jax.extend.backend jax.extend.core jax.extend.linear_util jax.extend.mlir jax.extend.random