burn-ndarray 0.9.0

Ndarray backend for the Burn framework
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56

[package]
authors = ["nathanielsimard <nathaniel.simard.42@gmail.com>"]
categories = ["science", "no-std", "embedded", "wasm"]
description = "Ndarray backend for the Burn framework"
edition = "2021"
keywords = ["deep-learning", "machine-learning", "data"]
license = "MIT OR Apache-2.0"
name = "burn-ndarray"
readme = "README.md"
repository = "https://github.com/burn-rs/burn/tree/main/burn-ndarray"
version = "0.9.0"

[features]
default = ["std"]

std = [
  "rand/std",
  "burn-tensor/std",
  "burn-common/std",
  "burn-autodiff",
  "rayon",
  "ndarray/std",
  "ndarray/rayon",
  "matrixmultiply/std",
  "matrixmultiply/threading",
]

blas-accelerate = ["ndarray/blas", "blas-src/accelerate"] # Accelerate framework (macOS only)
blas-netlib = ["ndarray/blas", "blas-src/netlib"]
blas-openblas = ["ndarray/blas", "blas-src/openblas", "openblas-src"]
blas-openblas-system = [
  "ndarray/blas",
  "blas-src/openblas",
  "openblas-src/system",
]

[dependencies]

# ** Please make sure all dependencies support no_std when std is disabled **

burn-autodiff = {path = "../burn-autodiff", version = "0.9.0", features = ["export_tests"], optional = true}
burn-common = {path = "../burn-common", version = "0.9.0", default-features = false}
burn-tensor = {path = "../burn-tensor", version = "0.9.0", default-features = false, features = ["export_tests"]}

matrixmultiply = {version = "0.3.7", default-features = false}
rayon = {workspace = true, optional = true}

blas-src = {version = "0.9.0", default-features = false, optional = true}# no-std compatible         

derive-new = {workspace = true}
libm = {workspace = true}
ndarray = {workspace = true}
num-traits = {workspace = true}
openblas-src = {version = "0.10.8", optional = true}
rand = {workspace = true}
spin = {workspace = true}# using in place of use std::sync::Mutex;