/tensorflow-nanoGPT

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

Primary LanguageJupyter NotebookMIT LicenseMIT

No issues in this repository yet.