mirror of
https://github.com/ROCm/jax.git
synced 2025-04-14 10:56:06 +00:00
34 lines
708 B
ReStructuredText
34 lines
708 B
ReStructuredText
.. _advanced_guide:
|
|
|
|
Advanced guides
|
|
===============
|
|
|
|
This section contains examples and tutorials on more advanced topics,
|
|
such as multi-core computation, automatic differentiation, and custom
|
|
operations.
|
|
|
|
.. toctree::
|
|
:caption: Parallel computation
|
|
:maxdepth: 1
|
|
|
|
notebooks/Distributed_arrays_and_automatic_parallelization
|
|
notebooks/explicit-sharding
|
|
notebooks/shard_map
|
|
multi_process
|
|
distributed_data_loading
|
|
|
|
.. toctree::
|
|
:caption: Automatic differentiation
|
|
:maxdepth: 1
|
|
|
|
notebooks/autodiff_cookbook
|
|
notebooks/Custom_derivative_rules_for_Python_code
|
|
notebooks/autodiff_remat
|
|
|
|
.. toctree::
|
|
:caption: Deep dives
|
|
:maxdepth: 1
|
|
|
|
notebooks/convolutions
|
|
xla_flags
|