poptorch
PopTorch是一个为Graphcore IPU开发的PyTorch扩展工具集,支持在IPU上进行模型训练、评估和使用。项目包含PopTorch和PopTorch Geometric两个主要组件,分别用于常规深度学习和图神经网络模型。项目提供了用户指南、安装说明和构建流程,主要适配Ubuntu 20.04环境。开发者可选择通过pip安装预编译wheel包或从源代码构建。PopTorch旨在让开发者充分利用IPU的计算能力,用于构建高性能AI应用。