hi-ml-azure>=0.1.8
pytorch-lightning>=1.4.9
torch>=1.8
