Back to Burn

Burn NdArray

crates/burn-ndarray/README.md

0.21.01.6 KB
Original Source

Burn NdArray

Burn ndarray backend

New projects should use burn-flex. It is a from-scratch pure-Rust CPU backend that replaces burn-ndarray with faster gemm, zero-copy view operations, native quantization, and full support for std, no_std, and WebAssembly. See burn-flex/COMPARISON.md for a migration path and operation-by-operation benchmarks.

Feature Flags

This crate can be used without the standard library (#![no_std]) with alloc by disabling the default std feature.

The following flags support various BLAS options:

  • blas-accelerate - Accelerate framework (macOS only)
  • blas-netlib - Netlib
  • blas-openblas - OpenBLAS static linked
  • blas-openblas-system - OpenBLAS from the system

Note: under the no_std mode, the seed is fixed if the seed is not initialized by Backend::seed method.

Platform Support

OptionCPUGPULinuxMacOSWindowsAndroidiOSWASM
Pure RustYesNoYesYesYesYesYesYes
AccelerateYesNoNoYesNoNoYesNo
NetlibYesNoYesYesYesNoNoNo
OpenblasYesNoYesYesYesYesYesNo