burn-tensor 0.22.0-pre.3

Tensor library with user-friendly APIs and automatic differentiation support
Documentation

burn-tensor

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

default

std (default)

colored (default)

accelerate

apple-amx

autodiff

autotune

autotune-checks

blas-netlib

capture

cpu

cubecl

cubecl-cpu

cubecl-cuda

cubecl-hip

cubecl-metal

cubecl-vulkan

cubecl-webgpu

cubecl-wgpu

cuda

doc

extension

This feature flag does not enable additional features.

flex

fusion

metal

ndarray

openblas

openblas-system

rayon

remote

remote-server

remote-websocket

rocm

simd

tch

template

tracing

vulkan

webgpu

wgpu

x86-v4