burn is a Rust deep learning framework that unifies training and inference through a single API, where the exact code used for training is the exact code that runs in production. It targets elimination of the export steps normally required to move models from Python training to production engines.
Project overview
By making the training codepath identical to the production inference codepath, burn removes the brittle export pipeline between Python training frameworks and deployment engines, which directly enables workflows such as on-device personalization and federated learning.
Project type
Model Development · Infrastructure
Deployment
Refer to project documentation
License
Apache-2.0
Best for
Teams with Rust proficiency, as implementing modules and running the framework strictly requires Rust knowledge, who need a single codepath for both model training and production inference without export steps.
AI engineers and researchers building models that require on-device personalization or federated learning, provided they are comfortable working in Rust.
Key capabilities
Executes multi-platform tensor operations via a single API where the exact code used for training is the exact code that runs in production.
Allows building burn with swappable backends, composing them to augment with functionalities like autodifferentiation and automatic kernel fusion.
Transparently equips a base backend with autodifferentiation support to call backward on models.
JIT-compiles streams of tensor operations to perform automatic kernel fusion, enhancing backends with improved speed.
Allows users to follow their training progress with a terminal UI dashboard, visualizing metrics in real-time without connecting to an external application.
Limitations and risks
ONNX support currently supports a limited set of ONNX operators.
There will be breaking changes as the project is in active development.
While issues from breaking changes are likely easy to fix, there are no guarantees at this stage.
Getting started
Install Rust, add the burn crate to Cargo.toml, define a neural network module, and run training using the Learner. Medium setup difficulty: requires Rust knowledge to implement modules and run the framework.
Evidence and sources
GitHub project description: Burn is a next generation tensor library and Deep Learning Framework that doesn't compromise on flexibility, efficiency and portability.
README: Burn unifies the two. By executing multi-platform tensor operations via a single, unified API, the exact code used for training is the exact code that runs in production.
README: Burn is both a tensor library and a deep learning framework, optimized for
README: Burn is currently in active development, and there will be breaking changes. While any resulting issues are likely to be easy to fix, there are no guarantees at this stage.
README: most code is generic over the Backend trait, which allows us to build Burn with swappable backends.