跳到正文
GitHub · 3D 生成新仓库·· 14 天前AI 评分42

Burn:用 Rust 统一训练与推理的开源深度学习框架

tracel-ai/burn

来源与原文

GitHub · 3D 生成新仓库

发布时间:(北京时间)

阅读原始来源 · github.com ↗
完整链接与内容说明

https://github.com/tracel-ai/burn

本站目前仅有 AI 导读与摘要,未展示原文正文;技术细节、代码和完整结论请查看原始来源。

AI 导读

tracel-ai 的开源 Rust 深度学习框架 Burn 通过统一 API 让训练与生产代码完全一致,支持端上个性化和联邦学习等场景。其基于 CubeCL 的后端覆盖 CUDA、ROCm、Metal、Vulkan、WebGPU 和 CPU,自动执行 kernel fusion;模型代码修改在 release 模式下增量编译不到 5 秒。LibTorch 已在 0.22.0 中废弃并从 main 移除。

来源:GitHub · 3D 生成新仓库 · github.com