Burn NdArray
Burn ndarray backend
New projects should use
burn-flex. It is a from-scratch pure-Rust CPU backend that replacesburn-ndarraywith faster gemm, zero-copy view operations, native quantization, and full support forstd,no_std, and WebAssembly. Seeburn-flex/COMPARISON.mdfor 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- Netlibblas-openblas- OpenBLAS static linkedblas-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
| Option | CPU | GPU | Linux | MacOS | Windows | Android | iOS | WASM |
|---|---|---|---|---|---|---|---|---|
| Pure Rust | Yes | No | Yes | Yes | Yes | Yes | Yes | Yes |
| Accelerate | Yes | No | No | Yes | No | No | Yes | No |
| Netlib | Yes | No | Yes | Yes | Yes | No | No | No |
| Openblas | Yes | No | Yes | Yes | Yes | Yes | Yes | No |