Kernel Hub
Hugging Face Kernel Hub 允许从 Hub 动态加载计算 kernel,支持 PyTorch 集成。kernel 可移植、版本唯一、兼容多 Python/PyTorch/CUDA 版本。用于加速模型推理和训练,支持自定义 kernel 发布到 Hub。
🎯适用场景:从 Hub 动态加载优化 kernel——加速推理/训练,支持自定义 kernel 发布
#计算优化#CUDA#PyTorch#Kernel#Hub 集成
📥 收录于 2026/8/23
📊 仓库数据
Stars553
Forks60
语言Python
更新2026/8/19
✅ 优点
- •553 stars 社区活跃
- •PyTorch 原生集成
- •支持多 CUDA 版本和 C++ ABI
- •Hub 生态可直接使用社区 kernel
⚠️ 限制
- •需要 torch>=2.5 和 CUDA
- •文档相对技术化
- •自定义 kernel 需要 Nix 构建环境
