jax
flax
optax
jaxopt
sympy
matplotlib
mpmath
numpy
