rust-ai-core 0.3.4

Unified AI engineering toolkit: orchestrates peft-rs, qlora-rs, unsloth-rs, axolotl-rs, bitnet-quantize, trit-vsa, vsa-optim-rs, and tritter-accel
Documentation
# THIS FILE IS AUTOMATICALLY GENERATED BY CARGO
#
# When uploading crates to the registry Cargo will automatically
# "normalize" Cargo.toml files for maximal compatibility
# with all versions of Cargo and also rewrite `path` dependencies
# to registry (e.g., crates.io) dependencies.
#
# If you are reading this file be aware that the original Cargo.toml
# will likely look very different (and much more reasonable).
# See Cargo.toml.orig for the original contents.

[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"