据AIbase网12月18日消息，美国谷歌公司正在推进一项名为“TorchTPU”的战略计划。该计划旨在显著提升其张量处理单元（TPU）芯片对PyTorch框架的兼容性，从而降低开发者从英伟达GPU切换至谷歌TPU的技术门槛和迁移成本。长期以来，谷歌的TPU主要适配自家的JAX框架，导致大量习惯使用PyTorch的开发者在调用TPU算力时面临性能瓶颈。通过TorchTPU项目，谷歌计划投入更多战略资源，优化底层软件与PyTorch的匹配度。此外，谷歌正考虑开源部分核心软件组件，以吸引更多开发者。据悉，谷歌还与PyTorch的主要维护方Meta展开深度合作，探讨让Meta获取更多TPU资源的可能性。