想亲手训练一个语言模型吗？本文将通过代码示例和实践指南，向您介绍如何在 TPU 上使用 JAX 从零构建并训练一个 GPT-2 模型，帮助您轻松入门。