torch>=2.3.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
numpy<2,>=1.23.0

[test]
absl-py
bitsandbytes==0.42.0
coverage==7.5.1
einops
expecttest==0.2.1
fdm==0.4.1
graphviz==0.20.3
hypothesis==6.104.2
jsonargparse
litgpt==0.3.1
numpy<2,>=1.23.0
pandas
pytest-benchmark==4.0.0
pytest-cov==5.0.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
transformers==4.42.4
xlsxwriter

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