Azure has full support for Tensorflow, Keras, PyTorch and the rest of the popular stuff. Shouldn't be a problem at all