mirror of
https://github.com/ROCm/jax.git
synced 2025-04-18 21:06:06 +00:00
44 lines
712 B
ReStructuredText
44 lines
712 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
|
|
|
|
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
|