/tensorflow-nanoGPT

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

Primary LanguageJupyter NotebookMIT LicenseMIT

Stargazers