mirror of
https://github.com/ROCm/jax.git
synced 2025-04-18 12:56:07 +00:00
18 lines
231 B
ReStructuredText
18 lines
231 B
ReStructuredText
.. currentmodule:: jax.extend
|
|
|
|
``jax.extend`` module
|
|
=====================
|
|
|
|
.. automodule:: jax.extend
|
|
|
|
Modules
|
|
-------
|
|
|
|
.. toctree::
|
|
:maxdepth: 1
|
|
|
|
jax.extend.core
|
|
jax.extend.linear_util
|
|
jax.extend.mlir
|
|
jax.extend.random
|