Burn NdArray
Burn ndarray backend
Deprecated: This crate is deprecated as of
0.22.0and will be removed in a future release. Please migrate to one of the actively maintained backends:
burn-flexfor portable pure-Rust CPU execution. It is a from-scratch backend that replacesburn-ndarraywith faster gemm, zero-copy view operations, native quantization, and full support forstd,no_std, and WebAssembly.- CubeCL backends (CUDA, ROCm, Vulkan, Metal, WebGPU) for GPU acceleration.
See
burn-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 |