16 releases (10 breaking)
Uses new Rust 2024
0.17.0 | Apr 23, 2025 |
---|---|
0.16.0 | Jan 14, 2025 |
0.15.0 | Oct 28, 2024 |
0.13.2 | May 3, 2024 |
0.8.0 | Jul 25, 2023 |
#883 in Machine learning
596 downloads per month
Used in 3 crates
1MB
23K
SLoC
burn-import
is a crate designed to simplify the process of importing models trained in other
machine learning frameworks into the Burn framework. This tool generates a Rust source file that
aligns the imported model with Burn's model and converts tensor data into a format compatible with
Burn.
Importing Models
The Burn project supports the import of models from various frameworks, emphasizing efficiency and compatibility. Currently, it handles two primary model formats:
-
ONNX: Facilitates direct import, ensuring the model's performance and structure are maintained.
-
PyTorch: Enables the loading of PyTorch model weights into Burn’s native model architecture, ensuring seamless integration.
Contribution
Interested in contributing to burn-import
? Check out our development guide for
more information.
Dependencies
~47–81MB
~1.5M SLoC