Lib.rs

Cargo Features

[dependencies]
burn-store = { version = "0.20.1", default-features = false, features = ["std", "memmap", "tracing", "burnpack", "wgpu", "metal", "cuda", "candle", "tch", "safetensors", "pytorch"] }

default = burnpack, memmap, pytorch, safetensors, std

These default features are set whenever burn-store is added without default-features = false somewhere in the dependency tree.

std default memmap

Enables memmap2 and regex, std of safetensors, std of burn-core, burn-tensor, and byteorder

Affects base::MAX_FILE_SIZE, store::FileStore…

memmap default = std

Enables memmap2

tracing

Enables tracing of optional burn-candle, burn-core, optional burn-cuda, burn-nn, optional burn-tch, burn-tensor, and optional burn-wgpu

burn-candle:

Optional backend dependencies for benchmarks

burnpack default = ciborium, serde

wgpu metal? = burn-wgpu

metal = wgpu

Enables metal of burn-wgpu

burn-wgpu:

Backends

cuda = burn-cuda

candle = burn-candle

tch = burn-tch

safetensors default std

Enables safetensors

pytorch default = serde, zip

Enables record-item-custom-serde of burn-core

burn-core:

Custom deserializer for Record that is helpful for importing data, such as PyTorch pt files.

Affects burn-store::pytorch…

Features from optional dependencies

In crates that don't use the dep: syntax, optional dependencies automatically become Cargo features. These features may have been created by mistake, and this functionality may be removed in the future.

ciborium burnpack

serde burnpack pytorch

zip pytorch

burn-candle candle?

burn-cuda cuda?

burn-tch tch?

burn-wgpu metal? wgpu?