使用Cloud Storage在Cloud TPU Pod上训练pytorch模型 免责声明：仅出于示例目的提供此代码。 此存储库包含支持文件，用于使用以下配置进行培训： 云TPU Pod TPU豆荚分为几片。 每个片都需要与VM Worker配对（异步训练）。 v2-32 TPU Pod具有4个切片（32/8 = 4）。 进行训练时，每个虚拟机将获得一个专用的pod切片。 XLA编译器执行代码转换，包括将矩阵平铺为较小的块，以有效地对矩阵单元（MXU）执行计算。 XLA编译器使用MXU硬