英伟达开源 PyCuTe:纯 Python 实现 GPU 核心布局代数

asdf1234_0 · x · 2026-07-30

NVIDIA 旗下 NVlabs 开源了 PyCuTe,这是 CUTLASS 3.x 和 CuTe DSL 核心的层级布局和张量代数的纯 Python 参考实现。

主要特点包括:

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →