英伟达开源 PyCuTe:纯 Python 实现 GPU 核心布局代数
asdf1234_0 · x · 2026-07-30
NVIDIA 旗下 NVlabs 开源了 PyCuTe,这是 CUTLASS 3.x 和 CuTe DSL 核心的层级布局和张量代数的纯 Python 参考实现。
主要特点包括:
- 无需 GPU 即可运行,方便在任何 Python 脚本中导入。
- 适用于学习代数原理、原型化新变换以及为 C++ 和 DSL 实现生成测试向量。
- 实现了 CuTe 白皮书中的各种布局代数运算,如 coalesce、composition、complement 和 logicalproduct 等。
「Infra」频道最新
- Reflection AI 开源 Beam:Apache 2.0 权重+40 余家生态伙伴 — zhyncs42 · 2026-10-16
- NVIDIA RTX Spark 正式发布,主打 Windows PC 本地跑 AI — ostrisai · 2026-10-16
- SkyRL 为 Beam 提供首日 RL 支持,LoRA 解码吞吐提升 62% — vllm_project · 2026-10-16
- DFlash2 为 Beam 解码提速最高 3.03 倍 — vllm_project · 2026-10-16
- Beam 长上下文服务优化:GB200 前向最快降 20% — vllm_project · 2026-10-16
- Reflection Beam 501B MoE 登陆 GB200:单用户 463 tok/s — zhyncs42 · 2026-10-16