user_guides.rst
.. _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
notebooks/thinking_in_jax
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
export/index
errors
transfer_guard
.. toctree::
:maxdepth: 1
:caption: Custom Operations
pallas/index