Transformations
===============

Master BrainState's state-aware program transformations, from the essential
``jit`` / ``grad`` / ``vmap`` through advanced batching, error handling, debugging,
and intermediate-representation tooling.

.. toctree::
   :maxdepth: 1

   01_jit_and_compilation
   02_autodiff
   03_vectorization
   04_advanced_batching
   05_control_flow
   06_error_handling_and_checks
   07_debugging
   08_ir_optimization_and_codegen