torch>=2.3.0
looseversion==1.3.0
lightning-utilities>=0.7.0
numpy
networkx>=3.3
optree>=0.12.1
opt_einsum>=3.3.0
mpmath<1.4.0
dill>=0.3.8

[notebooks]
cuda-python
ipython[all]~=8.30.0
liger-kernel==0.4.0
litgpt==0.5.1
numpy

[test]
absl-py
asv>=0.6.4
bitsandbytes==0.42.0
coverage~=7.6.8
einops
expecttest==0.3.0
fdm==0.5.0
graphviz==0.20.3
hypothesis~=6.124.7
jsonargparse
litgpt==0.4.11
numpy
pandas
pytest-benchmark==5.1.0
pytest-cov==6.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.48.2
xlsxwriter

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