rocm_jax/docs/user_guides.rst
jax authors afb775c168 Jax persistent compilation cache user guide.
This user guide covers using the cache on local filesystems
and Google Cloud.

PiperOrigin-RevId: 623236335
2024-04-09 11:48:33 -07:00

42 lines
667 B
ReStructuredText

.. _user-guides:
User Guides
===========
User guides are deeper dives into particular topics within JAX
that become relevant as your JAX project matures into larger
or deployed codebases.
.. toctree::
:maxdepth: 1
:caption: Debugging and Performance
profiling
device_memory_profiling
debugging/index
gpu_performance_tips
persistent_compilation_cache
.. toctree::
:maxdepth: 1
:caption: Development
jaxpr
notebooks/external_callbacks
type_promotion
pytrees
.. toctree::
:maxdepth: 1
:caption: Run Time
aot
errors
transfer_guard
.. toctree::
:maxdepth: 1
:caption: Custom Operations
pallas/index