_images/logo.svg

chaotaxΒΆ

Measure chaos in dynamical systems with JAX.

  • Differentiable, JIT-compatible, vectorizable Lyapunov exponent computation

  • Full spectrum via QR decomposition

  • Efficient implementation via forward-mode automatic differentiation

  • Works with any JAX-compatible stepper function

pip install chaotax

import jax, chaotax

exponent = chaotax.lyapunov(stepper)(u_0, key=jax.random.key(0))