Torchtpu

科技互联网

Google开源TorchTPU项目 为PyTorch提供原生TPU后端支持

Google已开源TorchTPU项目,为PyTorch框架提供了原生的张量处理单元(TPU)后端支持。该项目包含PyTorch ATen内核实现、编译缓存优化和分布式训练支持等功能,使用户能够在Google的TPU上直接运行PyTorch模型和操作。