relayrl_types 0.3.21

Data types for the RelayRL framework.
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

[package]
name = "relayrl_types"
version = "0.3.21"
authors = ["jrcalgo <jacksonr121@outlook.com>"]
edition = "2024"
description = "Data types for the RelayRL framework."
repository = "https://github.com/jrcalgo/relayrl"
documentation = "https://docs.rs/crate/relayrl_types"
readme = "README.md"
keywords = ['networking', 'learning', 'tensors', 'types', 'serialization']
categories = ['science', 'data-structures', 'encoding']
license = "Apache-2.0"

[features]
default = ["ndarray-backend", "codec-full", "inference-models"]
 
ndarray-backend = ["burn-ndarray"]
tch-backend = ["burn-tch"]

inference-models = ["tch-model", "onnx-model"]
tch-model = ["tch", "tokio", "tempfile"]
onnx-model = ["ort", "tokio", "tempfile", "ndarray"]

encryption = ["chacha20poly1305"]
compression = ["lz4_flex", "zstd"]
integrity = ["blake3"]
quantization = ["half"]
metadata = ["bincode"]
zerocopy = ["bytes"]

codec-basic = ["compression", "integrity", "zerocopy"]
codec-secure = ["codec-basic", "encryption"]
codec-full = ["codec-secure", "metadata", "quantization"]

[dependencies]
serde = { version = "1.0.228", features = ["derive"]}
serde_json = "1.0.145"

thiserror = "2.0.17"

bytemuck = "1.20.0"
dashmap = "6.1.0"
uuid = { version = "1.18.1", features = ["serde", "v4"] }

# TODO: Add support for these dependencies and remove tch entirely from relayrl_types
burn-tensor = "0.18.0"
burn-ndarray = { version = "0.18.0", optional = true}
burn-tch = { version = "0.18.0", optional = true}
# burn-rocm = {version = "0.18.0", optional = true} # TODO: Add support for rocm (probably 0.3.0)
# burn-cuda = {version = "0.18.0", optional = true} # TODO: Add support for cuda (probably 0.3.0)
# burn-wgpu = {version = "0.18.0", optional = true} # TODO: Add support for wgpu (probably 0.3.0)

tch = { version = "0.19.0", features = ["download-libtorch"], default-features = false, optional = true }
ort = { version = "2.0.0-rc.10", features = ["load-dynamic"], default-features = false, optional = true }
tokio = { version = "1.35.1", features = ["full"], optional = true }
tempfile = { version = "3.9.0", optional = true }
ndarray = { version = "0.15.6", optional = true }

[patch.crates-io]
torch-sys = "=0.19.0"

[dependencies.blake3]
version = "1.8.2"
optional = true

[dependencies.bytes]
version = "1.10.1"
optional = true

[dependencies.zstd]
version = "0.13.3"
optional = true

[dependencies.lz4_flex]
version = "0.11.5"
optional = true

[dependencies.chacha20poly1305]
version = "0.10.1"
optional = true

[dependencies.bincode]
version = "2.0.1"
features = ["serde"]
optional = true

[dependencies.half]
version = "2.7.0"
optional = true