[package]
edition = "2021"
rust-version = "1.92"
name = "rust-ai-core"
version = "0.3.4"
authors = ["Tyler Zervas <tz-dev@vectorweight.com>"]
build = false
autolib = false
autobins = false
autoexamples = false
autotests = false
autobenches = false
description = "Unified AI engineering toolkit: orchestrates peft-rs, qlora-rs, unsloth-rs, axolotl-rs, bitnet-quantize, trit-vsa, vsa-optim-rs, and tritter-accel"
homepage = "https://github.com/tzervas/rust-ai-core"
documentation = "https://docs.rs/rust-ai-core"
readme = "README.md"
keywords = [
"machine-learning",
"cuda",
"gpu",
"fine-tuning",
"quantization",
]
categories = [
"science",
"mathematics",
"development-tools",
]
license = "MIT"
repository = "https://github.com/tzervas/rust-ai-core"
[features]
cuda = [
"candle-core/cuda",
"dep:cubecl",
"dep:cubecl-cuda",
]
default = []
full = [
"cuda",
"python",
]
full-bindings = [
"cuda",
"python",
"napi",
"wasm",
]
napi = [
"dep:napi",
"dep:napi-derive",
]
python = [
"dep:pyo3",
"dep:numpy",
]
wasm = [
"dep:wasm-bindgen",
"dep:js-sys",
]
[lib]
name = "rust_ai_core"
crate-type = [
"rlib",
"cdylib",
]
path = "src/lib.rs"
[[example]]
name = "device_selection"
path = "examples/device_selection.rs"
[[example]]
name = "error_handling"
path = "examples/error_handling.rs"
[[example]]
name = "memory_tracking"
path = "examples/memory_tracking.rs"
[[test]]
name = "integration"
path = "tests/integration.rs"
[[bench]]
name = "core_ops"
path = "benches/core_ops.rs"
harness = false
[dependencies.axolotl-rs]
version = "1.1"
[dependencies.bitnet-quantize]
version = "0.1"
[dependencies.candle-core]
version = "0.9"
default-features = false
[dependencies.cubecl]
version = "0.9"
optional = true
[dependencies.cubecl-cuda]
version = "0.9"
optional = true
[dependencies.half]
version = "2.7"
[dependencies.js-sys]
version = "0.3"
optional = true
[dependencies.napi]
version = "2.16"
features = ["napi4"]
optional = true
default-features = false
[dependencies.napi-derive]
version = "2.16"
optional = true
[dependencies.numpy]
version = "0.27"
optional = true
[dependencies.peft-rs]
version = "1.0"
[dependencies.pyo3]
version = "0.27"
features = [
"extension-module",
"generate-import-lib",
]
optional = true
[dependencies.qlora-rs]
version = "1.0"
[dependencies.thiserror]
version = "2.0"
[dependencies.tracing]
version = "0.1"
[dependencies.tracing-subscriber]
version = "0.3"
features = ["env-filter"]
[dependencies.trit-vsa]
version = "0.1"
[dependencies.tritter-accel]
version = "0.2"
[dependencies.unsloth-rs]
version = "1.0"
[dependencies.vsa-optim-rs]
version = "0.1"
[dependencies.wasm-bindgen]
version = "0.2"
optional = true
[dev-dependencies.criterion]
version = "0.8"
[dev-dependencies.tokio]
version = "1.49"
features = ["full"]
[lints.clippy]
pedantic = "warn"
[lints.rust]
missing_docs = "warn"
unsafe_code = "deny"