技术实践
2026年PyTorch会议:谷歌与Meta共同推出了开源版本的TorchTPU,以便让PyTorch能够在TPU上顺利运行
在2026年北美PyTorch大会上,谷歌和Meta将TorchTPU开源了。TorchTPU是专为谷歌的TPU开发的PyTorch后端框架。Meta表示,计划利用这一工具将大型大语言模型训练任务转移到TPU上进行处理。同一周,AWS也为其Trainium平台发布了TorchNeuron。不过,开发者们对TPU的可用性以及当前这些技术是否会导致供应商锁定现象产生了质疑。 作者:Andrew Hoblitzell
pytorch tpu andrew