mirror of
https://github.com/ROCm/jax.git
synced 2025-04-16 11:56:07 +00:00
12eebfe8d4
* Create "extension guides" section * Sort developer notes into subsections * Move examples from advanced section into user guides * Reorder some listings, adjust some titles
47 lines
840 B
ReStructuredText
47 lines
840 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: Interfaces
|
|
|
|
pytrees
|
|
errors
|
|
aot
|
|
export/index
|
|
type_promotion
|
|
transfer_guard
|
|
|
|
.. toctree::
|
|
:maxdepth: 1
|
|
:caption: Custom operations
|
|
|
|
notebooks/external_callbacks
|
|
pallas/index
|
|
ffi
|
|
|
|
.. toctree::
|
|
:caption: Example applications
|
|
:maxdepth: 1
|
|
|
|
notebooks/neural_network_with_tfds_data
|
|
notebooks/Neural_Network_and_Data_Loading
|
|
notebooks/vmapped_log_probs
|