burn-tensor 0.19.0

Tensor library with user-friendly APIs and automatic differentiation support
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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118

[dependencies.burn-common]
default-features = false
version = "0.19.0"

[dependencies.burn-tensor-testgen]
optional = true
version = "0.19.0"

[dependencies.bytemuck]
features = ["extern_crate_alloc"]
version = "1.24.0"

[dependencies.colored]
optional = true
version = "3.0.0"

[dependencies.cubecl]
default-features = false
optional = true
version = "0.8.0"

[dependencies.cubecl-quant]
default-features = false
version = "0.8.0"

[dependencies.derive-new]
default-features = false
version = "0.7.0"

[dependencies.half]
default-features = false
features = ["alloc", "num-traits", "serde", "bytemuck"]
version = "2.7.1"

[dependencies.hashbrown]
version = "0.15.5"

[dependencies.num-traits]
default-features = false
features = ["libm"]
version = "0.2.19"

[dependencies.rand]
default-features = false
version = "0.9.2"

[dependencies.rand_distr]
default-features = false
version = "0.5.1"

[dependencies.serde]
default-features = false
features = ["derive", "alloc"]
version = "1.0.228"

[dependencies.serde_bytes]
default-features = false
features = ["alloc"]
version = "0.11.18"

[dev-dependencies.bincode]
default-features = false
features = ["alloc", "serde"]
version = "2.0.1"

[dev-dependencies.paste]
version = "1"

[dev-dependencies.rand]
default-features = false
features = ["thread_rng"]
version = "0.9.2"

[features]
cubecl = ["dep:cubecl"]
cubecl-cpu = ["cubecl", "cubecl/cpu"]
cubecl-cuda = ["cubecl", "cubecl/cuda"]
cubecl-hip = ["cubecl", "cubecl/hip"]
cubecl-wgpu = ["cubecl", "cubecl/wgpu"]
default = ["std", "burn-common/rayon"]
doc = ["default"]
experimental-named-tensor = []
export_tests = ["burn-tensor-testgen", "cubecl"]
std = ["rand/std", "half/std", "num-traits/std", "burn-common/std", "colored"]

[lib]
name = "burn_tensor"
path = "src/lib.rs"

[lints.clippy]

[lints.rustdoc]
broken_intra_doc_links = "deny"
invalid_html_tags = "deny"

[package]
authors = ["nathanielsimard <nathaniel.simard.42@gmail.com>"]
autobenches = false
autobins = false
autoexamples = false
autolib = false
autotests = false
build = false
categories = ["science", "no-std", "embedded", "wasm"]
description = "Tensor library with user-friendly APIs and automatic differentiation support"
documentation = "https://docs.rs/burn-tensor"
edition = "2024"
keywords = ["deep-learning", "machine-learning", "tensor", "pytorch", "ndarray"]
license = "MIT OR Apache-2.0"
name = "burn-tensor"
readme = "README.md"
repository = "https://github.com/tracel-ai/burn/tree/main/crates/burn-tensor"
resolver = "2"
version = "0.19.0"

[package.metadata.docs.rs]
features = ["doc"]
rustdoc-args = ["--cfg", "docsrs", "--html-in-header", "katex-header.html"]