mirror of
https://github.com/ROCm/jax.git
synced 2025-04-19 05:16:06 +00:00
2257e2075d
Co-authored-by: 8bitmp3 <19637339+8bitmp3@users.noreply.github.com> Co-authored-by: Jake VanderPlas <jakevdp@google.com>
20 lines
544 B
ReStructuredText
20 lines
544 B
ReStructuredText
.. _advanced_guide:
|
|
|
|
Advanced Tutorials
|
|
==================
|
|
|
|
.. toctree::
|
|
:maxdepth: 1
|
|
|
|
notebooks/autodiff_cookbook
|
|
multi_process
|
|
notebooks/Distributed_arrays_and_automatic_parallelization
|
|
notebooks/vmapped_log_probs
|
|
notebooks/neural_network_with_tfds_data
|
|
notebooks/Custom_derivative_rules_for_Python_code
|
|
notebooks/How_JAX_primitives_work
|
|
notebooks/Writing_custom_interpreters_in_Jax
|
|
notebooks/Neural_Network_and_Data_Loading
|
|
notebooks/xmap_tutorial
|
|
notebooks/external_callbacks
|
|
Custom_Operation_for_GPUs |