What do I have to do to ensure that a model can run on Gaudi?

Habana’s software integrates TensorFlow or PyTorch, allowing a compatible Python training script running on CPU or GPU to train a model on Gaudi, with the addition of just a few lines of code. See the Migration Guide for more information.