torch>=1.12.0
mpi4py>=3.1.0
hydra-core>=1.1.0
omegaconf>=2.1.0
pytz>=2021.1

[dev]
pytest>=6.0
black>=22.0
flake8>=4.0
mypy>=0.950

[intel]
intel-extension-for-pytorch
oneccl-bindings-for-pytorch
