burn 0.21.0

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 65 feature flags, 1 of them enabled by default.

default

std (default)

accelerate

apple-amx

audio

autodiff

autotune

autotune-checks

blas-netlib

burn-autodiff

burn-candle

burn-collective

burn-cpu

burn-cuda

burn-dispatch

burn-flex

burn-ir

burn-ndarray

burn-remote

burn-rocm

burn-router

burn-store

burn-tch

burn-train

burn-vision

burn-wgpu

candle

candle-cuda

candle-metal

collective

cpu

cubecl

cuda

dataset

dispatch

distributed

doc

flex

fusion

ir

metal

metrics

ndarray

network

openblas

openblas-system

record-item-custom-serde

remote

rl

rocm

router

server

simd

sqlite

sqlite-bundled

store

tch

template

tracing

train

tui

vision

vulkan

webgpu

wgpu

x86-v4