burn/burn-ndarray
Nathaniel Simard 9df20712bf
Fix docs (#1229)
2024-02-01 14:50:38 -05:00
..
src renaming FloatTensor Ops, Primitives, and maybe functions (#1174) 2024-01-27 10:04:50 -05:00
Cargo.toml Fix docs (#1229) 2024-02-01 14:50:38 -05:00
LICENSE-APACHE Refactor/backend ndarray (#105) 2022-11-18 20:37:38 -05:00
LICENSE-MIT Refactor/backend ndarray (#105) 2022-11-18 20:37:38 -05:00
README.md Chore/release (#1031) 2023-12-01 14:33:28 -05:00
build.rs Add accelerate framework support for burn-ndarray (#183) (#184) 2023-02-28 09:09:47 -05:00

README.md

Burn NdArray

Burn ndarray backend

Current Crates.io Version license

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 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