Burn Import
The burn-import crate enables seamless integration of pre-trained models from popular machine
learning frameworks into the Burn ecosystem. This functionality allows you to leverage existing
models while benefiting from Burn's performance optimizations and native Rust integration.
Supported Import Formats
Burn currently supports three primary model import formats, each serving different use cases:
| Format | Description | Use Case |
|---|---|---|
| ONNX (Guide) | Open Neural Network Exchange format | Direct import of complete model architectures and weights from any framework that supports ONNX export |
| PyTorch (Guide) | PyTorch weights (.pt, .pth) | Loading weights from PyTorch models into a matching Burn architecture |
| Safetensors (Guide) | Hugging Face's model serialization format | Loading a model's tensor weights into a matching Burn architecture |
ONNX Contributor Resources
- ONNX to Burn conversion guide - Instructions for adding support for additional ONNX operators
- ONNX tests README - Testing procedures for ONNX operators
- Supported ONNX Operators table - Complete list of currently supported ONNX operators