jax[cpu]
jax
numpy
scipy
matplotlib
pandas
optax
tomli-w
pyevtk
click
rich

[animation]
moviepy

[dev]
pytest
pytest-cov
coverage
ruff
python-dotenv

[docs]
sphinx
sphinx-rtd-theme
myst-parser
sphinx-autoapi
pandoc
jax[cpu]

[publishing]
build
twine
