mirror of
https://github.com/ROCm/jax.git
synced 2025-04-16 03:46:06 +00:00
21 lines
371 B
ReStructuredText
21 lines
371 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/Writing_custom_interpreters_in_Jax
|
||
|
|
jax.extend
|
||
|
|
|
||
|
|
.. toctree::
|
||
|
|
:caption: Libraries and extensions
|
||
|
|
:maxdepth: 1
|
||
|
|
|
||
|
|
building_on_jax
|