pydantic
omegaconf
pyyaml==5.3.1
nvidia-ml-py3
