burn

Burn 是一个下一代张量库和深度学习框架，旨在不牺牲灵活性、效率和可移植性。它使用 Rust 语言开发，支持自动微分、内核融合等高级特性，并通过静态图优化实现高性能计算。Burn 支持多种硬件后端，包括 CUDA、ROCm、Metal 和 Vulkan 等 GPU 后端，以及 X86 和 Arm 等 CPU 后端，确保在不同设备上都能高效运行。适用于需要跨平台模型训练和推理的场景，如云端训练与边缘设备部署相结合的应用。