numpy
scipy
jax
optax
flax
folx>=0.2.22
pyscf

[docs]
sphinx>=7.0
furo
myst-parser
sphinx-copybutton
sphinx-autodoc-typehints

[gpu]
jaxlib[cuda]

[test]
coverage[toml]>=7.0
