mirror of
https://github.com/ROCm/jax.git
synced 2025-04-17 04:16:07 +00:00
23 lines
413 B
ReStructuredText
23 lines
413 B
ReStructuredText
|
JAX reference documentation
|
||
|
|
===============================
|
||
|
|
|
||
|
|
Composable transformations of Python+NumPy programs: differentiate, vectorize,
|
||
|
|
JIT to GPU/TPU, and more
|
||
|
|
|
||
|
|
For an introduction to JAX, start at the
|
||
|
|
`JAX GitHub page <https://github.com/google/jax>`_.
|
||
|
|
|
||
|
|
.. toctree::
|
||
|
|
:maxdepth: 3
|
||
|
|
:caption: Contents:
|
||
|
|
|
||
|
|
jax
|
||
|
|
|
||
|
|
|
||
|
|
Indices and tables
|
||
|
|
==================
|
||
|
|
|
||
|
|
* :ref:`genindex`
|
||
|
|
* :ref:`modindex`
|
||
|
|
* :ref:`search`
|