numpy
jaxlib
jax
optax
sympy
pyyaml
jupyterlab
matplotlib
ipywidgets
