Google与Meta开源TorchTPU架构

发生了什么

据arXiv cs.AI报道，Google与Meta联合开源了TorchTPU架构，旨在实现PyTorch在Google TPU上的原生执行。该架构通过混合急切/编译执行方式，提供无缝模型执行和TPU迁移，同时借助急切执行优化、torch.compile集成及原生硬件加速，实现行业竞争力的推理和训练性能。