:orphan:
:nosearch:

Superseded documentation
========================

The pages listed here are the previous generation of the JAX documentation.
They have been superseded by the leveled documentation — :doc:`101 <101/index>`
through :doc:`501 <501/index>` — but remain available at their original URLs
so that existing links keep working. They are no longer maintained; prefer
the current documentation.

.. toctree::
   :maxdepth: 1

   jax-101
   key-concepts
   notebooks/thinking_in_jax
   beginner_guide
   notebooks/shard_map
   notebooks/layout
   multi_process
   fault_tolerance
   security
   distributed_data_loading
   notebooks/autodiff_cookbook
   notebooks/autodiff_remat
   advanced_autodiff
   debugging
   debugging/index
   transfer_guard
   persistent_compilation_cache
   buffer_donation
   benchmarking
   profiling
   device_memory_profiling
   array_refs
   external-callbacks
   ffi
   notebooks/cute_dsl_jax
   gradient-checkpointing
   aot
   export/index
   xla_flags
   hijax_types
   async_dispatch
   concurrency
   gpu_memory_allocation
   errors
   export/jax2tf
   notebooks/neural_network_with_tfds_data
   notebooks/Neural_Network_and_Data_Loading
   notebooks/vmapped_log_probs