burn-optim 0.19.1

Optimizer building blocks for the Burn deep learning 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
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
119
120
121
122
123

[dependencies.burn-collective]
default-features = false
optional = true
version = "0.19.1"

[dependencies.burn-core]
default-features = false
version = "0.19.1"

[dependencies.burn-cuda]
default-features = false
optional = true
version = "0.19.1"

[dependencies.burn-fusion]
optional = true
version = "0.19.1"

[dependencies.burn-remote]
default-features = false
optional = true
version = "0.19.1"

[dependencies.burn-rocm]
default-features = false
optional = true
version = "0.19.1"

[dependencies.burn-router]
default-features = false
optional = true
version = "0.19.1"

[dependencies.burn-tch]
optional = true
version = "0.19.1"

[dependencies.burn-wgpu]
default-features = false
optional = true
version = "0.19.1"

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

[dependencies.hashbrown]
features = ["serde"]
version = "0.15.5"

[dependencies.log]
default-features = false
optional = true
version = "0.4.28"

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

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

[dev-dependencies.burn-autodiff]
version = "0.19.1"

[dev-dependencies.burn-ndarray]
version = "0.19.1"

[dev-dependencies.burn-nn]
version = "0.19.1"

[dev-dependencies.rstest]
version = "0.26.1"

[features]
collective = ["burn-collective"]
default = ["std", "burn-core/default"]
doc = ["std", "burn-core/doc"]
std = ["burn-core/std", "num-traits/std", "serde/std", "log"]
test-cuda = ["burn-cuda/default"]
test-memory-checks = ["burn-fusion/memory-checks"]
test-metal = ["test-wgpu", "burn-wgpu/metal"]
test-rocm = ["burn-rocm/default"]
test-tch = ["burn-tch/default"]
test-vulkan = ["test-wgpu", "burn-wgpu/vulkan"]
test-wgpu = ["burn-wgpu/default"]

[lib]
name = "burn_optim"
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 = "Optimizer building blocks for the Burn deep learning framework"
documentation = "https://docs.rs/burn-optim"
edition = "2024"
keywords = ["deep-learning", "machine-learning", "tensor", "pytorch", "ndarray"]
license = "MIT OR Apache-2.0"
name = "burn-optim"
readme = "README.md"
repository = "https://github.com/tracel-ai/burn/tree/main/crates/burn-optim"
resolver = "2"
version = "0.19.1"

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