[package]
name = "ml-supervised"
version.workspace = true
edition.workspace = true
rust-version.workspace = true
authors.workspace = true
license.workspace = true
repository.workspace = true
homepage.workspace = true
documentation.workspace = true
publish.workspace = true
keywords.workspace = true
categories.workspace = true
description = "Supervised models (TFT, Mamba, Liquid, TGGN, TLOB, KAN, xLSTM, Diffusion)"

[features]
default = ["cuda"]
cuda = []

[dependencies]
ml-core.workspace = true
common.workspace = true
config.workspace = true
data.workspace = true

# Async
tokio.workspace = true

# GPU: direct cudarc + cuBLAS for all models (candle eliminated)
cudarc = { version = "0.19", default-features = false, features = [
    "driver", "cublas", "dynamic-linking", "std", "cuda-version-from-build-system",
] }

# Safetensors for checkpoint save/load (direct, not via candle wrapper)
safetensors = "0.7"

# Serialization
serde = { workspace = true, features = ["derive"] }
serde_json.workspace = true

# Core utilities
thiserror.workspace = true
anyhow.workspace = true
tracing.workspace = true
rand.workspace = true
uuid.workspace = true
async-trait.workspace = true

# BF16 support
half.workspace = true

# Numerics
ndarray = { workspace = true, features = ["rayon"] }
nalgebra = { version = "0.33", features = ["serde-serialize"] }

# Graph support (TGGN)
petgraph = { version = "0.6", features = ["serde"] }

# Caching (TFT)
lru.workspace = true

# System
libc = "0.2"
num_cpus = "1.16"

# Concurrency
rayon.workspace = true
dashmap = { workspace = true }
parking_lot = { version = "0.12", features = ["hardware-lock-elision"] }

[dev-dependencies]
tokio = { workspace = true, features = ["test-util", "macros"] }
approx.workspace = true
tempfile = "3"

[lints]
workspace = true