unknown

Burn is a deep learning framework and tensor library written in Rust that prioritizes flexibility, efficiency, and portability. It supports multiple backends including CUDA, ROCm, Metal, Vulkan, WebGPU, and CPU implementations across various hardware platforms. The framework features swappable backends through a Backend trait, automatic differentiation via backend decorators, kernel fusion for optimization, and ONNX model import. Burn enables training with a built-in TUI dashboard, supports no_std environments for embedded devices, and can run inference in browsers via WebAssembly. The framework aims to eliminate the Python/C++ dichotomy common in ML frameworks by leveraging Rust's zero-cost abstractions.

tracel-ai/burn: Burn is a next generation tensor library and Deep Learning Framework that doesn't compromise on flexibility, efficiency and portability.

Become a cool developer with Dev Source! Your ultimate dev source for resources, insights, and a thriving community to learn, grow, and stay ahead every single day!