mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-19 05:16:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/docs/advanced_guide.rst

34 lines
708 B
ReStructuredText
Raw Normal View History

Update doc landing page Co-authored-by: 8bitmp3 <19637339+8bitmp3@users.noreply.github.com> Co-authored-by: Jake VanderPlas <jakevdp@google.com>
2023-01-13 09:59:40 -08:00
.. _advanced_guide:
docs: tidy up titles and headings 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.
2024-08-12 22:13:36 -07:00
Advanced guides
===============
This section contains examples and tutorials on more advanced topics,
docs: reorganize sections * Create "extension guides" section * Sort developer notes into subsections * Move examples from advanced section into user guides * Reorder some listings, adjust some titles
2024-08-13 13:16:50 -07:00
such as multi-core computation, automatic differentiation, and custom
operations.
DOC: organize content in advanced guide
2023-01-25 15:12:04 -08:00
.. toctree::
docs: sentence case index and sub-index headings We currently use both forms, so for consistency (and easier reading), pick this one.
2024-08-11 17:48:19 -07:00
:caption: Parallel computation
Update doc landing page Co-authored-by: 8bitmp3 <19637339+8bitmp3@users.noreply.github.com> Co-authored-by: Jake VanderPlas <jakevdp@google.com>
2023-01-13 09:59:40 -08:00
:maxdepth: 1
notebooks/Distributed_arrays_and_automatic_parallelization
Explicit sharding docs
2025-03-13 11:54:52 -04:00
notebooks/explicit-sharding
[shard-map] add user tutorial
2024-01-08 13:12:04 -08:00
notebooks/shard_map
docs: reorganize sections * Create "extension guides" section * Sort developer notes into subsections * Move examples from advanced section into user guides * Reorder some listings, adjust some titles
2024-08-13 13:16:50 -07:00
multi_process
Add Distributed data loading doc
2024-05-10 20:36:17 +00:00
distributed_data_loading
DOC: organize content in advanced guide
2023-01-25 15:12:04 -08:00
.. toctree::
docs: sentence case index and sub-index headings We currently use both forms, so for consistency (and easier reading), pick this one.
2024-08-11 17:48:19 -07:00
:caption: Automatic differentiation
DOC: organize content in advanced guide
2023-01-25 15:12:04 -08:00
:maxdepth: 1
notebooks/autodiff_cookbook
Update doc landing page Co-authored-by: 8bitmp3 <19637339+8bitmp3@users.noreply.github.com> Co-authored-by: Jake VanderPlas <jakevdp@google.com>
2023-01-13 09:59:40 -08:00
notebooks/Custom_derivative_rules_for_Python_code
fixes from reviewers
2023-02-22 09:26:43 -08:00
notebooks/autodiff_remat
DOC: organize content in advanced guide
2023-01-25 15:12:04 -08:00
.. toctree::
docs: sentence case index and sub-index headings We currently use both forms, so for consistency (and easier reading), pick this one.
2024-08-11 17:48:19 -07:00
:caption: Deep dives
DOC: organize content in advanced guide
2023-01-25 15:12:04 -08:00
:maxdepth: 1
fixes from reviewers
2023-02-22 09:26:43 -08:00
notebooks/convolutions
Add md page about XLA&XLA_FLAGS in JAX
2024-08-18 19:24:44 -07:00
xla_flags
Reference in New Issue Copy Permalink