burn 0.22.0-pre.4

Flexible and Comprehensive Deep Learning Framework in Rust
Documentation

burn

There is currently very little information to present on this page because Docs.rs has only limited support for extracting structured feature metadata from Cargo crates. This issue is tracked in Rust RFC #3416. Check this library's main docs, readme, and Cargo.toml in case its authors have documentation for features available there instead.

This version has 53 feature flags, 5 of them enabled by default.

default

optim (default)

rl (default)

std (default)

autodiff (default)

burn-optim (default)

accelerate

apple-amx

audio

autotune

autotune-checks

blas-netlib

burn-store

burn-train

burn-vision

capture

cpu

cubecl

cuda

dataset

doc

extension

flex

fusion

ir

linalg

metal

metrics

ndarray

network

openblas

openblas-system

pytorch

rayon

remote

remote-server

remote-websocket

rocm

safetensors

signal

simd

sqlite

sqlite-bundled

store

tch

template

tracing

train

tui

vision

vulkan

webgpu

wgpu

x86-v4