jax

[demo]
matplotlib
tqdm
ratinabox
pykalman
ipykernel

[dev]
pytest>=7.0
pytest-cov
build
twine
ruff
