80 lines
1.9 KiB
TOML
80 lines
1.9 KiB
TOML
[package]
|
|
name = "rtx-model-merging"
|
|
version = "1.0.0"
|
|
edition.workspace = true
|
|
rust-version = "1.92"
|
|
authors = ["RustyTorch Team"]
|
|
license = "MIT OR Apache-2.0"
|
|
description = "Advanced model merging algorithms for RustyTorch"
|
|
repository = "https://github.com/rustytorch/rustytorch"
|
|
keywords = ["machine-learning", "model-merging", "ties", "dare", "slerp"]
|
|
categories = ["machine-learning", "algorithms"]
|
|
|
|
[dependencies]
|
|
# Workspace dependencies
|
|
rtx-tensor = { workspace = true }
|
|
rtx-runtime = { workspace = true }
|
|
rtx-memory = { workspace = true }
|
|
rtx-validation = { workspace = true }
|
|
rtx-autograd = { workspace = true }
|
|
rtx-losses = { workspace = true }
|
|
|
|
# Essential
|
|
anyhow = { workspace = true }
|
|
thiserror = { workspace = true }
|
|
tracing = { workspace = true }
|
|
serde = { workspace = true }
|
|
serde_json = { workspace = true }
|
|
tokio = { workspace = true }
|
|
|
|
# Math and computation
|
|
nalgebra = { workspace = true }
|
|
num-complex = { workspace = true }
|
|
rand = { workspace = true }
|
|
rand_distr = { workspace = true }
|
|
statrs = { workspace = true }
|
|
ndarray = { workspace = true }
|
|
|
|
# Collections and utilities
|
|
indexmap = { workspace = true }
|
|
dashmap = { workspace = true }
|
|
uuid = { workspace = true }
|
|
sha2 = { workspace = true }
|
|
bincode = { workspace = true }
|
|
|
|
# Async and concurrency
|
|
futures = { workspace = true }
|
|
async-trait = { workspace = true }
|
|
parking_lot = { workspace = true }
|
|
crossbeam = { workspace = true }
|
|
rayon = { workspace = true }
|
|
|
|
# Configuration
|
|
config = { workspace = true }
|
|
toml = { workspace = true }
|
|
|
|
# GPU acceleration
|
|
cudarc = { workspace = true, optional = true }
|
|
|
|
# Time
|
|
chrono = { workspace = true }
|
|
|
|
[dev-dependencies]
|
|
tokio-test = { workspace = true }
|
|
proptest = { workspace = true }
|
|
criterion = { workspace = true }
|
|
tempfile = { workspace = true }
|
|
|
|
[features]
|
|
default = []
|
|
gpu = ["cudarc"]
|
|
metal = ["rtx-tensor/metal", "rtx-runtime/metal", "rtx-memory/metal"]
|
|
experimental = []
|
|
|
|
|
|
[lib]
|
|
name = "rtx_model_merging"
|
|
path = "src/lib.rs"
|
|
[lints]
|
|
workspace = true
|