ztensor 1.2.2

Unified, zero-copy, and safe I/O for deep learning formats
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

[package]
name = "ztensor"
version = "1.2.2"
edition = "2021"
authors = ["In Gim <in.gim@icloud.com>"]
description = "Unified, zero-copy, and safe I/O for deep learning formats"
license = "MIT"
repository = "https://github.com/pie-project/ztensor"

[lib]
crate-type = ["cdylib", "rlib"]

[features]
default = []
safetensors = ["dep:safetensors"]
pickle = ["dep:zip"]
gguf = []
npz = ["dep:zip"]
onnx = []
hdf5 = ["dep:flate2"]
all-formats = ["safetensors", "pickle", "gguf", "npz", "onnx", "hdf5"]
cffi = ["dep:libc"]
python = ["dep:pyo3", "dep:numpy", "all-formats"]

[dependencies]
serde = { version = "1.0", features = ["derive"] }
ciborium = "0.2"
thiserror = "2"
zstd = "0.13"
crc32c = "0.6.8"
libc = { version = "0.2", optional = true }
memmap2 = "0.9"
bytemuck = { version = "1.16", features = ["derive"] }
half = { version = "2.4", features = ["bytemuck"] }
sha2 = "0.10"
hex = "0.4"
safetensors = { version = "0.4", optional = true }
zip = { version = "2", optional = true, default-features = false, features = ["deflate"] }
flate2 = { version = "1", optional = true }
pyo3 = { version = "0.24", features = ["extension-module"], optional = true }
numpy = { version = "0.24", optional = true }

[profile.release]
lto = "fat"
codegen-units = 1

[dev-dependencies]
tempfile = "3.8"
safetensors = "0.4"
zip = { version = "2", default-features = false, features = ["deflate"] }
bytemuck = { version = "1.16", features = ["derive"] }
half = { version = "2.4", features = ["bytemuck"] }