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

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

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