ExBurn — Elixir bridge to the Burn deep learning framework.

ExBurn provides a high-level API for tensor computation, neural network training, and GPU-accelerated machine learning by delegating to Burn via Rust NIFs (Native Implemented Functions).

Architecture

Elixir/Axon → Nx.Defn → ExBurn.Defn.Compiler → ExBurn.Backend → ExBurn.Nif (Rustler) → Burn/CubeCL → GPU
                                                                                 ↕
                                                                           ExCubecl (GPU buffers, kernels, pipelines)

Quick Start

# Set ExBurn as the default Nx backend
Nx.default_backend(ExBurn.Backend)

# Create and manipulate tensors
t = Nx.tensor([1.0, 2.0, 3.0])
Nx.add(t, t) |> Nx.to_list()

Modules

• ExBurn.Defn.Compiler — Nx.Defn.Compiler implementation for GPU-accelerated defn
• ExBurn.Backend — Nx backend that delegates to Burn via NIF
• ExBurn.Nif — Rustler NIF stubs for Burn interop
• ExBurn.Tensor — Tensor conversion utilities between Nx and Burn formats
• ExBurn.BurnBridge — High-level bridge for Burn operations and ExCubecl buffers
• ExBurn.CubeclBridge — GPU compute via ExCubecl (buffers, kernels, pipelines)
• ExBurn.Model — Model definition and training orchestration
• ExBurn.Training — Training loop implementation
• ExBurn.Serving — Nx.Serving integration for batched concurrent inference