burn-tensor 0.22.0-pre.2

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

default

std (default)

colored (default)

autodiff

autotune

autotune-checks

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

remote

remote-server

remote-websocket

rocm

tch

tracing

vulkan

webgpu

wgpu