torch>=2.2.0
lightning==2.3.0.dev20240428
jsonargparse[signatures]>=4.27.6

[all]
bitsandbytes==0.42.0
sentencepiece>=0.2.0
tokenizers>=0.15.2
requests>=2.31.0
litdata>=0.2.2
litserve>=0.1.0
zstandard>=0.22.0
pandas>=1.9.0
pyarrow>=15.0.2
tensorboard>=2.14.0
torchmetrics>=1.3.1
datasets>=2.18.0
transformers>=4.38.0
lm-eval>=0.4.2
safetensors>=0.4.3
huggingface_hub[hf_transfer]>=0.21.0

[test]
pytest>=8.1.1
pytest-rerunfailures>=14.0
pytest-timeout>=2.3.1
pytest-dependency>=0.6.0
transformers>=4.38.0
einops>=0.7.0
protobuf>=4.23.4

[test:python_version >= "3.10"]
lightning-thunder==0.2.0.dev20240505
