burn

Burn 是一个基于 Rust 的新一代张量库与深度学习框架，专注于在灵活性、计算效率和跨平台可移植性之间取得平衡。它支持自动微分、内核融合，并提供统一的 Backend trait 抽象，可无缝切换 CUDA、ROCm、Metal、Vulkan、WebGPU、Wasm 等多种硬件后端，同时兼容 CPU（含 no-std 环境）与 LibTorch。其设计允许在训练、推理及边缘部署等不同阶段复用同一套模型代码，特别适用于需多平台适配、低延迟部署或嵌入式/浏览器环境（如 WebAssembly）的 AI 应用场景。