batch_forge/Cargo.toml at main · yash27-lab/batch_forge · GitHub

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
[package]
name = "batch_forge"
version = "0.1.0"
edition = "2021"
rust-version = "1.75"
description = "Correctness-first Metal inference runtime for JAX/Equinox models, written in Rust"
license = "MIT OR Apache-2.0"
repository = "https://github.com/yash27-lab/batch_forge"
keywords = ["inference", "metal", "apple-silicon", "jax", "llm"]
categories = ["science", "hardware-support"]

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

[[bin]]
name = "batch_forge"
path = "src/main.rs"

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

[dependencies]
safetensors = "0.4"
memmap2 = "0.9"
serde_json = "1"
bytemuck = { version = "1.14", features = ["derive"] }
thiserror = "1.0"
tracing = "0.1"
tracing-subscriber = { version = "0.3", features = ["env-filter"] }
tokio = { version = "1", features = ["rt-multi-thread", "macros", "sync"] }

[target.'cfg(target_os = "macos")'.dependencies]
metal = "0.28"

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