Advanced Automatic Differentiation# Advanced automatic differentiation Higher-order derivatives Forward- and reverse-mode autodiff in JAX Complex numbers and differentiation Custom derivative rules for JAX-transformable Python functions