rust_trainer 0.1.4

CPU-first pure-Rust supervised trainer for Selective State Space Models with Hyperspherical Prototype Networks.
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

[package]
name = "rust_trainer"
version = "0.1.4"
edition = "2021"
description = "CPU-first pure-Rust supervised trainer for Selective State Space Models with Hyperspherical Prototype Networks."
authors = ["NeuroMamba Contributors"]
license = "Apache-2.0"
repository = "https://github.com/neuromamba/rust_trainer"
homepage = "https://github.com/neuromamba/rust_trainer"
documentation = "https://docs.rs/rust_trainer"
readme = "README.md"
keywords = ["mamba", "ssm", "trainer", "cpu", "hpn"]
categories = ["science", "algorithms", "command-line-utilities"]
exclude = ["runs/", "target/"]

[lib]
name = "rust_trainer"
path = "src/lib.rs"
crate-type = ["cdylib", "rlib"]

[[bin]]
name = "train_generic"
path = "src/bin/train_generic.rs"

[dependencies]
ndarray = { version = "0.17.2", features = ["rayon", "serde"] }
rand = "0.10.1"
rand_distr = "0.6.0"
rayon = "1.12.0"
wide = "1.3.0"
serde = { version = "1.0.228", features = ["derive"] }
serde_json = "1.0.149"
serde_yaml = "0.9.34"
bincode = { version = "2.0.1", features = ["serde"] }
pyo3 = { version = "0.22", features = ["extension-module"], optional = true }

[features]
default = []
python = ["pyo3"]

[profile.release]
opt-level = 3
lto = "fat"
codegen-units = 1
panic = "abort"