jax
tqdm
jaxtyping

[:python_version < "3.11"]
typing-extensions

[notebooks]
ipykernel
ipython
jupyterlab
matplotlib

[test]
pytest
pytest-cov
pytest-mock
chex
