lightning[app]>=2.1.0
torch>=1.10.0
numpy

[all]
deepspeed>=0.6.0
tqdm
tiktoken
datasets

[deepspeed]
deepspeed>=0.6.0

[dev]
deepspeed>=0.6.0
tqdm
tiktoken
datasets
lightning>=1.8.4
pytest-cov==4.0.0
codecov==2.1.13
coverage==6.5.0
pytest==7.2.0

[nanogpt]
tqdm
tiktoken
datasets

[test]
lightning>=1.8.4
pytest-cov==4.0.0
codecov==2.1.13
coverage==6.5.0
pytest==7.2.0
