torch>=2.2.0
looseversion==1.3.0
lightning-utilities>=0.7.0
numpy<2,>=1.23.0
igraph>=0.10.4
optree>=0.9.2
opt_einsum>=3.3.0
mpmath<1.4.0

[notebooks]
cuda-python
ipython[all]==8.23.0

[test]
absl-py
coverage==7.4.4
einops
expecttest==0.2.1
fdm==0.4.1
graphviz==0.20.1
hypothesis==6.100.0
jsonargparse
litgpt==0.3.0
numpy
pandas
pytest-cov==4.1.0
pytest-random-order==1.1.1
pytest-timeout==2.2.0
pytest-timestamper==0.0.10
pytest-xdist==3.5.0
pytest==8.1.1
xlsxwriter

[test:sys_platform == "linux" or sys_platform == "darwin"]
jax
jaxlib
