31 lines
1.2 KiB
TOML
31 lines
1.2 KiB
TOML
[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
|