Example how to train GPT-2 (XLA + AMP), export to SavedModel and serve with Tensorflow Serving - View it on GitHub