cacb1f8874
Dependency crates all compile with BF16: - ml-dqn: noisy_layers, target_update, gpu_replay_buffer, branching → BF16 - ml-ppo: stubbed cuda_compile usage, PPO ops return errors (cold path) - ml-supervised: liquid training host data → BF16 conversion - ml-ensemble, ml-explainability: stubbed cuda_compile Remaining: 108 errors in ml crate itself (Phase 3 Task 14 continuing). Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
78 lines
1.8 KiB
TOML
78 lines
1.8 KiB
TOML
[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
|