airML/Cargo.toml at master · rlaope/airML · 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
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
[workspace]
resolver = "2"
members = [
    "crates/airml-core",
    "crates/airml-preprocess",
    "crates/airml-providers",
    "crates/airml-embed",
]

[package]
name = "airml"
version = "0.1.0"
edition = "2021"
description = "Lightweight ML runtime - Run ML models without Python"
license = "MIT"
repository = "https://github.com/airml/airml"
keywords = ["machine-learning", "onnx", "inference", "rust"]
categories = ["science", "command-line-utilities"]

[dependencies]
# Workspace crates
airml-core = { path = "crates/airml-core" }
airml-preprocess = { path = "crates/airml-preprocess" }
airml-providers = { path = "crates/airml-providers" }
airml-embed = { path = "crates/airml-embed" }

# CLI
clap = { version = "4.5", features = ["derive"] }

# Error handling
anyhow = "1.0"

# Async (for future use)
tokio = { version = "1.43", features = ["rt", "macros"], optional = true }

[features]
default = ["cpu"]
cpu = ["airml-providers/cpu"]
coreml = ["airml-providers/coreml"]
nlp = ["airml-preprocess/nlp"]

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

[profile.dev]
opt-level = 1

[workspace.dependencies]
# ONNX Runtime
ort = { version = "2.0.0-rc.11", default-features = false }

# Tensor/Array
ndarray = "0.16"
half = "2.4"

# Image processing
image = { version = "0.25", default-features = false, features = ["png", "jpeg"] }
nshare = "0.10"

# Error handling
thiserror = "2.0"
anyhow = "1.0"