[package]
edition = "2024"
name = "burn-store"
version = "0.20.1"
authors = ["Dilshod Tadjibaev (@antimora)"]
build = false
autolib = false
autobins = false
autoexamples = false
autotests = false
autobenches = false
description = "Storage and serialization infrastructure for Burn"
documentation = "https://docs.rs/burn-store"
readme = "README.md"
keywords = [
"deep-learning",
"machine-learning",
"tensor",
"storage",
"serialization",
]
categories = [
"science",
"no-std",
"embedded",
"wasm",
]
license = "MIT OR Apache-2.0"
repository = "https://github.com/tracel-ai/burn/tree/main/crates/burn-store"
resolver = "2"
[features]
burnpack = [
"serde",
"ciborium",
]
candle = ["burn-candle"]
cuda = ["burn-cuda"]
default = [
"std",
"pytorch",
"safetensors",
"burnpack",
"memmap",
]
memmap = [
"std",
"dep:memmap2",
]
metal = [
"wgpu",
"burn-wgpu/metal",
]
pytorch = [
"burn-core/record-item-custom-serde",
"zip",
"serde",
]
safetensors = ["dep:safetensors"]
std = [
"dep:memmap2",
"safetensors/std",
"burn-core/std",
"burn-tensor/std",
"dep:regex",
"byteorder/std",
]
tch = ["burn-tch"]
tracing = [
"burn-candle?/tracing",
"burn-core/tracing",
"burn-cuda?/tracing",
"burn-nn/tracing",
"burn-tch?/tracing",
"burn-tensor/tracing",
"burn-wgpu?/tracing",
]
wgpu = ["burn-wgpu"]
[lib]
name = "burn_store"
path = "src/lib.rs"
[[example]]
name = "burnpack_inspect"
path = "examples/burnpack_inspect.rs"
[[bench]]
name = "resnet18_loading"
path = "benches/resnet18_loading.rs"
harness = false
[[bench]]
name = "unified_loading"
path = "benches/unified_loading.rs"
harness = false
[[bench]]
name = "unified_saving"
path = "benches/unified_saving.rs"
harness = false
[[bench]]
name = "zero_copy_loading"
path = "benches/zero_copy_loading.rs"
harness = false
[dependencies.burn-candle]
version = "0.20.1"
optional = true
[dependencies.burn-core]
version = "0.20.1"
default-features = false
[dependencies.burn-cuda]
version = "0.20.1"
optional = true
[dependencies.burn-nn]
version = "0.20.1"
default-features = false
[dependencies.burn-tch]
version = "0.20.1"
optional = true
[dependencies.burn-tensor]
version = "0.20.1"
default-features = false
[dependencies.burn-wgpu]
version = "0.20.1"
optional = true
[dependencies.byteorder]
version = "1.5.0"
default-features = false
[dependencies.bytes]
version = "1.10"
default-features = false
[dependencies.ciborium]
version = "0.2"
optional = true
default-features = false
[dependencies.half]
version = "2.7.1"
features = [
"alloc",
"num-traits",
"serde",
]
default-features = false
[dependencies.hashbrown]
version = "0.16"
features = ["serde"]
[dependencies.memmap2]
version = "0.9"
optional = true
[dependencies.regex]
version = "1.12.1"
features = [
"perf",
"unicode",
]
optional = true
default-features = false
[dependencies.safetensors]
version = "0.7.0"
optional = true
default-features = false
[dependencies.serde]
version = "1.0.228"
features = [
"derive",
"alloc",
]
optional = true
default-features = false
[dependencies.textdistance]
version = "1.1.1"
default-features = false
[dependencies.zip]
version = "7.0.0"
optional = true
[dev-dependencies.burn-ndarray]
version = "0.20.1"
[dev-dependencies.divan]
version = "0.1"
[dev-dependencies.tempfile]
version = "3.24.0"
[target.'cfg(not(target_has_atomic = "ptr"))'.dependencies.bytes]
version = "1.10"
features = ["extra-platforms"]
default-features = false
[lints.clippy]
[lints.rustdoc]
broken_intra_doc_links = "deny"
invalid_html_tags = "deny"