mirror of
https://github.com/ROCm/jax.git
synced 2025-04-16 03:46:06 +00:00
3c223cd253
This shortens some titles and makes them more consistent. It also
removes "JAX" from several titles ("in JAX", "for JAX", "JAX's",
etc.). Since these are JAX docs, that ought to be clear from context.
48 lines
1009 B
ReStructuredText
48 lines
1009 B
ReStructuredText
.. _advanced_guide:
|
|
|
|
Advanced guides
|
|
===============
|
|
|
|
This section contains examples and tutorials on more advanced topics,
|
|
such as multi-core computation, custom operations, and more in-depth
|
|
applications.
|
|
|
|
.. toctree::
|
|
:caption: Examples
|
|
:maxdepth: 1
|
|
|
|
notebooks/neural_network_with_tfds_data
|
|
notebooks/Neural_Network_and_Data_Loading
|
|
notebooks/vmapped_log_probs
|
|
|
|
.. toctree::
|
|
:caption: Parallel computation
|
|
:maxdepth: 1
|
|
|
|
multi_process
|
|
notebooks/Distributed_arrays_and_automatic_parallelization
|
|
notebooks/shard_map
|
|
distributed_data_loading
|
|
|
|
.. toctree::
|
|
:caption: Automatic differentiation
|
|
:maxdepth: 1
|
|
|
|
notebooks/autodiff_cookbook
|
|
notebooks/Custom_derivative_rules_for_Python_code
|
|
notebooks/autodiff_remat
|
|
|
|
.. toctree::
|
|
:caption: JAX internals
|
|
:maxdepth: 1
|
|
|
|
notebooks/How_JAX_primitives_work
|
|
notebooks/Writing_custom_interpreters_in_Jax
|
|
Custom_Operation_for_GPUs
|
|
|
|
.. toctree::
|
|
:caption: Deep dives
|
|
:maxdepth: 1
|
|
|
|
notebooks/convolutions
|