brainharmony 0.1.0

Brain-Harmony multimodal brain foundation model — inference in Rust with Burn ML
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
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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115

[package]
name = "brainharmony"
version = "0.1.0"
edition = "2021"
rust-version = "1.78"
description = "Brain-Harmony multimodal brain foundation model — inference in Rust with Burn ML"
keywords      = ["fmri", "foundation-model", "neuroscience", "brain", "vision-transformer"]
categories    = ["science", "mathematics", "algorithms"]
authors       = ["Eugene Hauptmann"]
license       = "MIT"
homepage      = "https://github.com/eugenehp/brainharmony-rs"
repository    = "https://github.com/eugenehp/brainharmony-rs"
documentation = "https://docs.rs/brainharmony"
readme        = "README.md"

exclude = [
    "data/*",
    "figures/*",
    "hf_model_card/*",
]

[lib]
name = "brainharmony"
path = "src/lib.rs"

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

[[example]]
name = "embed"
path = "examples/embed.rs"

[[example]]
name = "batch"
path = "examples/batch.rs"
required-features = ["ndarray"]

[[example]]
name = "classify"
path = "examples/classify.rs"
required-features = ["ndarray"]

[[example]]
name = "csv_export"
path = "examples/csv_export.rs"
required-features = ["ndarray"]

[[example]]
name = "profile"
path = "examples/profile.rs"
required-features = ["ndarray"]

[[example]]
name = "bench"
path = "examples/bench.rs"
required-features = ["ndarray"]

[[example]]
name = "bench_gpu"
path = "examples/bench_gpu.rs"

[features]
default = ["ndarray"]
# CPU backend — Rayon multi-threading + SIMD enabled by default
ndarray         = ["dep:burn-ndarray", "burn/ndarray"]
# macOS: swap matmul impl to Apple Accelerate (multi-threaded, very fast on Apple Silicon)
# Recommended: cargo build --release --features accelerate
accelerate      = ["blas-accelerate"]
blas-accelerate = ["burn-ndarray?/blas-accelerate"]
# Linux: use system OpenBLAS (apt install libopenblas-dev)
openblas-system = ["burn-ndarray?/blas-openblas-system"]
# GPU backend — wgpu on Metal (macOS) or Vulkan (Linux with GPU)
wgpu            = ["burn/wgpu", "dep:burn-cubecl", "dep:burn-backend", "dep:cubek", "dep:cubecl"]
# GPU backend with f16 (half-precision) — 2x less memory, faster on supported GPUs.
# Requires Metal (macOS) or Vulkan (Linux). Not compatible with ndarray (CPU).
# Usage: cargo build --release --no-default-features --features wgpu-f16
wgpu-f16        = ["burn/wgpu", "dep:burn-cubecl", "dep:burn-backend", "dep:cubek", "dep:cubecl"]
# HuggingFace Hub downloader — requires TLS
hf-download     = ["dep:hf-hub"]

[dependencies]
burn = { version = "0.20.1", default-features = false, features = ["std", "simd"] }
burn-ndarray = { version = "0.20.1", default-features = false, features = ["std", "simd", "multi-threads"], optional = true }
burn-cubecl = { version = "0.20.1", optional = true }
burn-backend = { version = "0.20.1", optional = true }
cubek = { version = "0.1.1", optional = true }
cubecl = { version = "0.9.0", optional = true }

safetensors = "0.7"
half = { version = "2", features = ["bytemuck"] }
rayon = "1"
clap = { version = "4", features = ["derive", "env"] }
serde = { version = "1", features = ["derive"] }
serde_json = "1"
serde_yaml = "0.9"
anyhow = "1"
fastrand = "2"
thiserror = "2"
ndarray = "0.16"

# HuggingFace Hub client — optional; only compiled with --features hf-download.
hf-hub = { version = "0.5", default-features = false, features = ["ureq"], optional = true }

[dev-dependencies]
tempfile = "3"

[profile.release]
opt-level = 3
lto = "thin"
codegen-units = 1
strip = true

[patch.crates-io]
cubek-matmul = { git = "https://github.com/eugenehp/cubek.git", branch = "cubek-matmul", package = "cubek-matmul" }