numpy>=1.17.2
torch>=1.11.0
tqdm>=4.57.0
PyYAML>=5.4
fsspec[http]>2021.06.0
torchmetrics>=0.7.0
packaging>=17.1
typing-extensions>=4.0.0
lightning-utilities>=0.6.0.post0

[all]
matplotlib>3.1
omegaconf>=2.0.5
hydra-core>=1.0.5
jsonargparse[signatures]>=4.18.0
rich!=10.15.0.a,>=10.14.0
tensorboardX>=2.2
deepspeed>=0.6.0
lightning-colossalai==0.1.0dev1
torchvision>=0.11.1
gym[classic_control]>=0.17.0
ipython[all]<8.7.1

[deepspeed]
deepspeed>=0.6.0

[dev]
matplotlib>3.1
omegaconf>=2.0.5
hydra-core>=1.0.5
jsonargparse[signatures]>=4.18.0
rich!=10.15.0.a,>=10.14.0
tensorboardX>=2.2
deepspeed>=0.6.0
lightning-colossalai==0.1.0dev1
torchvision>=0.11.1
gym[classic_control]>=0.17.0
ipython[all]<8.7.1
coverage==6.5.0
codecov==2.1.12
pytest==7.2.0
pytest-cov==4.0.0
pytest-forked==1.4.0
pytest-rerunfailures==10.3
pre-commit==2.20.0
cloudpickle>=1.3
scikit-learn>0.22.1
onnxruntime<1.14.0
psutil<5.9.5
pandas>1.0
fastapi<0.87.0
uvicorn<0.19.1
tensorboard>=2.9.1
protobuf<=3.20.1

[examples]
torchvision>=0.11.1
gym[classic_control]>=0.17.0
ipython[all]<8.7.1

[extra]
matplotlib>3.1
omegaconf>=2.0.5
hydra-core>=1.0.5
jsonargparse[signatures]>=4.18.0
rich!=10.15.0.a,>=10.14.0
tensorboardX>=2.2

[lightning-colossalai]
lightning-colossalai==0.1.0dev1

[strategies]
deepspeed>=0.6.0
lightning-colossalai==0.1.0dev1

[test]
coverage==6.5.0
codecov==2.1.12
pytest==7.2.0
pytest-cov==4.0.0
pytest-forked==1.4.0
pytest-rerunfailures==10.3
pre-commit==2.20.0
cloudpickle>=1.3
scikit-learn>0.22.1
onnxruntime<1.14.0
psutil<5.9.5
pandas>1.0
fastapi<0.87.0
uvicorn<0.19.1
tensorboard>=2.9.1
protobuf<=3.20.1
