lightning>=1.8.0
torch>=1.10.0
numpy

[all]
deepspeed>=0.6.0
tiktoken
datasets
tqdm

[deepspeed]
deepspeed>=0.6.0

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

[nanogpt]
tiktoken
datasets
tqdm

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