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/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