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.25.0

[test]
absl-py
coverage==7.5.1
einops
expecttest==0.2.1
fdm==0.4.1
graphviz==0.20.3
hypothesis==6.103.0
jsonargparse
litgpt==0.3.1
numpy
pandas
pytest-benchmark==4.0.0
pytest-cov==4.1.0
pytest-random-order==1.1.1
pytest-timeout==2.3.1
pytest-timestamper==0.0.10
pytest-xdist==3.6.1
pytest==8.1.1
xlsxwriter

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