numpy
scipy

[all]
jax
jaxlib
matplotlib
plotly
optax

[autodiff]
jax
jaxlib
optax

[plot]
matplotlib
plotly
