numpy
jax
optax
equinox

[notebook]
jupyter
matplotlib
