Google与Meta开源TorchTPU,为谷歌TPU提供原生PyTorch后端,支持即时执行、编译执行和分布式计算。Google称其可减少代码改动,并在TPUv7上展示了大模型推理与训练性能;项目仍在完善动态形状、编译性能等功能。
来源:InfoQ
本站内容来自 Telegram 频道,实时更新
订阅频道