huggingface-hub<1.4,>=0.30
lightning>=2.6.1
psutil==7.2.2
safetensors>=0.4.3
tokenizers>=0.21
torch>=2.7
tqdm>4.66

[:python_version >= "3.10"]
jsonargparse[signatures]<=4.41,>=4.37

[compiler]

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

[extra]
datasets<4,>=2.18
huggingface-hub[hf-transfer]>=0.21
litdata==0.2.59
litlogger>=0.1.7
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.57,>=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.50,>=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
