Part 3: Transformations
=======================

This section covers JAX transformations and program optimization in BrainState.

.. toctree::
   :maxdepth: 1

   01_jit_compilation
   02_automatic_differentiation
   03_vectorization
   04_parallelization.ipynb
   05_loops_conditions
   06_other_transforms