[package]
name = "rtx-training"
version = "1.0.0"
edition.workspace = true
rust-version = "1.92"
authors = ["RustyTorch Team"]
license = "MIT OR Apache-2.0"
repository = "https://github.com/rustytorch/rustytorch"
description = "RTX training and optimization stack - Everything needed for ML training"
keywords = ["machine-learning", "deep-learning", "training", "distributed", "transformers"]
categories = ["science", "mathematics"]

[dependencies]
# Core
rtx-tensor = { path = "../../core/rtx-tensor", version = "1.0.0" }
rtx-runtime = { path = "../../core/rtx-runtime", version = "1.0.0" }
rtx-autograd = { path = "../../core/rtx-autograd", version = "1.0.0" }

# Training essentials
rtx-transformers = { path = "../../training/rtx-transformers", version = "1.0.0" }
# rtx-distributed = { path = "../../training/rtx-distributed", version = "1.0.0" }  # Temporarily disabled - missing hpc-channels
rtx-flash-attention = { path = "../../training/rtx-flash-attention", version = "1.0.0" }
rtx-rl = { path = "../../training/rtx-rl", version = "1.0.0" }
rtx-compress = { path = "../../training/rtx-compress", version = "1.0.0" }

[dev-dependencies]
tokio-test = "0.4"
criterion = "0.5"
[lints]
workspace = true