开源项目

pytorch/pytorch:统一图捕获与重放抽象,重塑可复用执行API

pytorch/pytorch

AI 基础设施AI 开发与运维平台

概述

PyTorch 在框架层推进图捕获与图重放机制的统一,把分散在各执行入口的捕获逻辑收敛到一套抽象之下。对开发者而言,模型完成捕获后可在不同场景重放执行,可复用执行 API 的形态和编写模型代码的方式随之改变;依赖底层执行细节的推理引擎、编译器与分布式框架需要跟进接口调整。本次公开材料未披露具体接口清单,建议以仓库提交与文档为准。