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

[dev]
pytest
pytest-regressions
pytest-cov
black
codespell
pep8-naming
pre-commit
pydocstyle
pyright

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

[notebooks]
jupyter
ipykernel
