mirror of
https://github.com/ROCm/jax.git
synced 2025-04-17 04:16:07 +00:00
22 lines
400 B
ReStructuredText
22 lines
400 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
|
|
Custom_Operation_for_GPUs
|
|
jax.extend
|
|
|
|
.. toctree::
|
|
:caption: Libraries and extensions
|
|
:maxdepth: 1
|
|
|
|
building_on_jax
|