base_image: mcr.microsoft.com/azureml/openmpi4.1.0-ubuntu20.04:latest