mirror of
https://github.com/ROCm/jax.git
synced 2025-04-16 20:06:05 +00:00
12eebfe8d4
* Create "extension guides" section * Sort developer notes into subsections * Move examples from advanced section into user guides * Reorder some listings, adjust some titles
24 lines
446 B
ReStructuredText
24 lines
446 B
ReStructuredText
.. _extensions:
|
|
|
|
Extension guides
|
|
================
|
|
|
|
Guides for extending JAX's capabilities, and for building libraries
|
|
that use or interface with JAX.
|
|
|
|
.. toctree::
|
|
:caption: Extensible JAX internals
|
|
:maxdepth: 1
|
|
|
|
notebooks/How_JAX_primitives_work
|
|
jaxpr
|
|
notebooks/Writing_custom_interpreters_in_Jax
|
|
Custom_Operation_for_GPUs
|
|
jax.extend
|
|
|
|
.. toctree::
|
|
:caption: Libraries and extensions
|
|
:maxdepth: 1
|
|
|
|
building_on_jax
|