tracel-ai/burn

Health v2 · maintenance only

Adoption not part of the score

15816 stars · 1026 forks observed · 2026-08-28

What it is AI-extracted, prompt v1, taxonomy v1, 2026-08-29, confidence not recorded

Burn is a Rust-based tensor library and deep learning framework supporting training and inference through a unified API. It JIT-compiles tensor operations with automatic kernel fusion, running on CUDA, Metal, WebGPU, Vulkan, ROCm, and CPU across servers, desktops, and mobile devices.

Use cases

• train neural networks in rust
• run deep learning inference on any device
• deploy models without python export step
• on-device model personalization and federated learning
• run models in the browser with webgpu and wasm
• port pytorch models via onnx to rust
• gpu kernel fusion for faster inference

When to choose

• you want a single rust codebase for training and production inference
• you need portable deployment across gpu vendors, mobile, and wasm
• you want pytorch-like ergonomics with compiled performance

When to avoid

• your team and tooling are python-centric and you don't need rust
• you rely on python-only ecosystem features like certain training libraries
• you need the broadest model zoo support available today

Facets

library · maturity active

machine-learning deep-learning llm-training llm-inference gpu-computing wasm deep-learning machine-learning gpu-computing cross-platform rust wasm cross-platform tensor-library autodiff kernel-fusion onnx webgpu pytorch-alternative scientific-computing cuda gpu

Member repositories

For agents

markdown · JSON · MCP: product_card(name="tracel-ai/burn")