2019-01-15 20:14:19 -05:00
|
|
|
JAX reference documentation
|
|
|
|
|
===============================
|
|
|
|
|
|
|
|
|
|
Composable transformations of Python+NumPy programs: differentiate, vectorize,
|
2019-06-04 10:09:43 -04:00
|
|
|
JIT to GPU/TPU, and more.
|
2019-01-15 20:14:19 -05:00
|
|
|
|
|
|
|
|
For an introduction to JAX, start at the
|
|
|
|
|
`JAX GitHub page <https://github.com/google/jax>`_.
|
|
|
|
|
|
2019-09-30 11:00:02 -07:00
|
|
|
.. toctree::
|
|
|
|
|
:maxdepth: 1
|
|
|
|
|
:caption: Tutorials
|
|
|
|
|
|
|
|
|
|
notebooks/quickstart
|
|
|
|
|
notebooks/autodiff_cookbook
|
2019-10-03 11:20:04 +02:00
|
|
|
|
|
|
|
|
.. toctree::
|
|
|
|
|
:maxdepth: 1
|
|
|
|
|
:caption: Advanced Tutorials
|
|
|
|
|
|
2019-09-30 11:00:02 -07:00
|
|
|
notebooks/Common_Gotchas_in_JAX
|
2019-10-02 14:41:28 +02:00
|
|
|
notebooks/How_JAX_primitives_work
|
2019-09-30 11:00:02 -07:00
|
|
|
|
2019-06-04 10:09:43 -04:00
|
|
|
.. toctree::
|
|
|
|
|
:maxdepth: 1
|
|
|
|
|
:caption: Notes
|
|
|
|
|
|
|
|
|
|
async_dispatch
|
2019-07-23 09:53:27 -04:00
|
|
|
concurrency
|
2019-07-29 12:24:58 -07:00
|
|
|
gpu_memory_allocation
|
2019-08-08 21:02:41 -04:00
|
|
|
profiling
|
2019-08-25 14:28:53 -07:00
|
|
|
rank_promotion_warning
|
2019-06-04 10:09:43 -04:00
|
|
|
|
2019-01-15 20:14:19 -05:00
|
|
|
.. toctree::
|
|
|
|
|
:maxdepth: 3
|
2019-09-30 11:00:02 -07:00
|
|
|
:caption: API documentation
|
2019-01-15 20:14:19 -05:00
|
|
|
|
|
|
|
|
jax
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
Indices and tables
|
|
|
|
|
==================
|
|
|
|
|
|
|
|
|
|
* :ref:`genindex`
|
|
|
|
|
* :ref:`modindex`
|
|
|
|
|
* :ref:`search`
|
