lightning>=1.8.0
torch>=1.10.0
numpy

[all]
deepspeed>=0.6.0
tiktoken
tqdm
datasets

[deepspeed]
deepspeed>=0.6.0

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

[nanogpt]
tiktoken
tqdm
datasets

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