jax
jaxlib
sympy
numpy
attrs

[dev]
plotly
kaleido
jraph
flax
dm-haiku
equinox
optax
tqdm
pytest
nox
s2fft
