Burn Core

Modules, configuration, records and data loading for Burn

Applications should depend on burn, which re-exports everything here together with neural network layers, optimizers and training. This crate is the dependency for libraries that only need the core abstractions.

• tensor: Burn's tensor API, re-exported from burn-tensor.
• module: the Module trait, implemented with #[derive(Module)], and Param for trainable tensors.
• config: serializable configuration structs with #[derive(Config)].
• store: module records and the burnpack format.
• data: datasets, batchers and data loaders (std only).
• prelude: the types most programs import.

Feature Flags

Backend features (wgpu, cuda, flex, ...) and the autodiff, fusion, remote and capture features match those of burn. Others:

• std (default): standard library support. Without it the crate is no_std with alloc.
• dataset: the dataset library in data; vision, audio and sqlite add sources.
• network: file downloads with a progress bar.
• tracing: instrument operations with the tracing crate.

Part of the Burn deep learning framework. See the Burn Book and the API documentation.