dm-haiku>0.0.9
h5py
hydra-core
jax
jaxlib
jax-dataclasses
kfac-jax
scipy
optax
pyscf
tensorboardX
pyyaml
tqdm
uncertainties
chex<1.0.0

[dev]
pytest
pytest-regressions
black
codespell
flake8
flake8-bugbear
flake8-comprehensions
flake8-quotes
isort
mypy
pep8-naming
pre-commit
pydocstyle
pyright

[doc]
sphinx
Jinja2
sphinxcontrib-katex
toml
docutils
pydata-sphinx-theme
