burn 0.22.0-pre.2

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, 6 of them enabled by default.

default

optim (default)

rl (default)

std (default)

autodiff (default)

burn-optim (default)

burn-autodiff (default)

accelerate

apple-amx

audio

autotune

autotune-checks

blas-netlib

burn-candle

burn-cpu

burn-cuda

burn-dispatch

burn-flex

burn-ir

burn-ndarray

burn-rocm

burn-router

burn-store

burn-tch

burn-train

burn-vision

burn-wgpu

candle

candle-cuda

candle-metal

cpu

cubecl

cuda

dataset

doc

extension

flex

fusion

ir

metal

metrics

ndarray

network

openblas

openblas-system

pytorch

remote

remote-server

remote-websocket

rocm

router

safetensors

simd

sqlite

sqlite-bundled

store

tch

template

tracing

train

tui

vision

vulkan

webgpu

wgpu

x86-v4