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

[all]
bitsandbytes==0.42.0
sentencepiece
tokenizers
datasets
requests
litdata
zstandard
pandas
pyarrow
tensorboard
torchmetrics
huggingface_hub[hf_transfer]>=0.21.0

[test]
pytest
pytest-rerunfailures
pytest-timeout
transformers>=4.38.0
einops
protobuf
