tracel-ai/burn

Burn is a deep learning framework and tensor library written in Rust. A tensor is the basic data type used in machine learning, a multi-dimensional array of numbers. A deep learning framework gives you the tools to build, train, and run neural networks on top of those tensors. PyTorch and TensorFlow are the two best-known frameworks; Burn is in the same category but is built around Rust rather than Python. A central design choice is that Burn is generic over what it calls the Backend trait. The same model code can run on many different pieces of hardware without rewriting it. The README lists supported GPU backends including CUDA for Nvidia, ROCm for AMD, Metal for Apple, Vulkan, WebGPU, and LibTorch as a bridge to the PyTorch C++ runtime. CPU backends include a CubeCL-based path for x86 and ARM, a backend called Flex that also targets WebAssembly and no-std, and the LibTorch CPU path. Backends can be wrapped with decorators that add capabilities. Autodiff adds automatic backpropagation, the math step that lets a model learn from its mistakes; a short Rust example in the README shows how wrapping the Wgpu backend with Autodiff makes the backward call available. Fusion is another decorator that merges many small GPU operations into a single fused kernel for speed, and it is enabled by default on accelerated backends. A Router decorator can split work across multiple backends so some operations run on the CPU and others on the GPU. A Remote decorator has a client-server split that sends tensor operations to a remote compute machine over a network connection. Burn is designed to cover both training and inference. The README mentions an ergonomic dashboard for monitoring training progress and says models can be deployed everywhere from embedded devices to large GPU clusters. It also says the transition from training to deployment requires no code changes, unlike some Python frameworks. The project is maintained by tracel-ai under the MIT and Apache 2.0 licenses, with a Discord server, documentation at burn.dev/docs/burn, and a Crates.io release for the burn crate.