Modules
Application callback for ExBurn.
Nx backend implementation that delegates tensor operations to Burn via NIF.
High-level bridge for Burn operations.
Bridge to ExCubecl for GPU execution via Burn's CubeCL backend.
Dataset utilities for ExBurn.
Custom Nx.Defn.Compiler that compiles defn expressions
to run on the Burn GPU backend via NIF.
Structured error type for ExBurn operations.
Model definition and training orchestration for ExBurn.
Helper module that wraps NIF calls and returns {:ok, result} tuples.
Nx.Serving integration for ExBurn.
Nx.Serving callback implementation for ExBurn.
Tensor conversion utilities between Nx and Burn formats.
Training loop implementation for ExBurn models.
Saves model checkpoints at specified intervals.
Stops training when validation loss stops improving.
Records all training metrics into a history list.
Logs training metrics after each epoch.
Reduces learning rate when validation loss stops improving.
Learning rate warmup callback.