[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