burn-std 0.20.0-pre.6

Core types and utilities shared across the Burn ecosystem.
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

[package]
authors = ["Dilshod Tadjibaev (@antimora)"]
categories = []
description = "Core types and utilities shared across the Burn ecosystem."
documentation = "https://docs.rs/burn-std"
edition.workspace = true
keywords = []
license.workspace = true
name = "burn-std"
readme.workspace = true
repository = "https://github.com/tracel-ai/burn/tree/main/crates/burn-std"
version.workspace = true

[lints]
workspace = true

[features]
cubecl = ["dep:cubecl"]
default = ["std", "cubecl-common/default"]
doc = ["default"]
network = ["dep:indicatif", "dep:reqwest", "dep:tokio"]
std = ["cubecl-common/std", "num-traits/std"]

[dependencies]
bytemuck = { workspace = true, features = ["extern_crate_alloc"] }
half = { workspace = true, features = ["bytemuck"] }
num-traits = { workspace = true }
serde = { workspace = true }

cubecl = { workspace = true, optional = true, default-features = false }
cubecl-common = { workspace = true, default-features = false, features = [
    "serde",
    "shared-bytes",
] }
# Enable extra-platforms for portable-atomic support on targets without native atomics (e.g., thumbv6m)
# This is needed because cubecl-common's shared-bytes feature pulls in bytes
bytes = { workspace = true }

# Network downloader
indicatif = { workspace = true, optional = true }
reqwest = { workspace = true, optional = true }
tokio = { workspace = true, optional = true }

[dev-dependencies]
dashmap = { workspace = true }

# Enable extra-platforms for bytes on targets without native atomics (e.g., thumbv6m-none-eabi)
[target.'cfg(not(target_has_atomic = "ptr"))'.dependencies]
bytes = { workspace = true, features = ["extra-platforms"] }

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