英伟达开源 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」频道最新
- TensorSharp 引擎新增多 GPU 张量并行,显著提升本地 GGUF 推理速度 — fuzhongkai · 2026-07-30
- UC Berkeley 提出 K-search:将 CUDA 内核优化自动迁移至苹果 MLX — berkeley_ai · 2026-07-30
- 搭载 Nvidia Thor 算力,Advantech 边缘设备跑通 GMSL 摄像头 — chrismatthieu · 2026-07-30
- OpenRouter 数据:Together 提供 Kimi K3 最低价与最高缓存命中率 — zhyncs42 · 2026-07-30
- 三星Q2营业利润飙升1800%创新高,AI芯片需求强劲 — Polymarket · 2026-07-30
- AI 算力竞赛背后的隐忧:全球电力产能扩张现状盘点 — lemire · 2026-07-30