dm-haiku
e3nn-jax
h5py
hydra-core
jax[cuda]
jaxtyping
kfac-jax
pyscf
tensorboard
pyyaml
tqdm
uncertainties

[dev]
pytest
pytest-regressions
black
flake8<6.0.0
flake8-bugbear
flake8-comprehensions
flake8-quotes
isort
pep8-naming
pydocstyle
