改一个词就能把AI代码搬到谷歌芯片上。

谷歌和Meta周二开源TorchTPU,让PyTorch直接跑在谷歌的TPU芯片上。已有代码只要把设备名从cuda改成tpu就能迁移。Meta已用它跑广告和语言模型,亚马逊也发布了针对自家芯片的同类工具。

为什么重要?英伟达的护城河,一部分来自只在它芯片上好用的软件。代码一旦这么容易搬动,你用的那些服务就能换上更便宜的芯片。

相关新闻