jax
flax
optax
sympy
matplotlib
mpmath
numpy
