pydantic
omegaconf
pyyaml==5.4.1
nvidia-ml-py3
