huggingface-hub<0.33,>=0.23.5
lightning>=2.5
psutil==7
safetensors>=0.4.3
tokenizers>=0.15.2
torch>=2.5
tqdm>=4.66

[:python_version <= "3.9"]
jsonargparse[signatures]<=4.32.1,>=4.30.1

[:python_version > "3.9"]
jsonargparse[signatures]>=4.37

[compiler]

[compiler:python_version >= "3.10" and sys_platform == "linux"]
lightning-thunder>=0.2.0.dev20250119

[extra]
datasets<4,>=2.18
huggingface-hub[hf-transfer]>=0.21
litdata==0.2.51
litserve>0.2
lm-eval!=0.4.9.1,>=0.4.2
pandas>=1.9
pyarrow>=15.0.2
requests>=2.31
sentencepiece>=0.2
tensorboard>=2.14
torchmetrics>=1.3.1
transformers<4.54,>=4.51.3
zstandard>=0.22

[extra:sys_platform != "win32"]
uvloop>=0.2

[extra:sys_platform == "darwin"]
bitsandbytes<0.43,>=0.42

[extra:sys_platform == "linux" or sys_platform == "win32"]
bitsandbytes<0.45.5,>=0.45.2

[test]
einops>=0.7
protobuf>=4.23.4
pytest>=8.1.1
pytest-benchmark>=5.1
pytest-dependency>=0.6
pytest-rerunfailures>=14
pytest-timeout>=2.3.1
