jax
sympy
scipy
