54 lines
2.0 KiB
TOML
54 lines
2.0 KiB
TOML
[package]
|
|
name = "rtx"
|
|
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"
|
|
documentation = "https://docs.rs/rtx"
|
|
readme = "README.md"
|
|
description = "Complete RTX ML/AI framework - The world's most comprehensive and performant ML platform"
|
|
keywords = ["machine-learning", "deep-learning", "artificial-intelligence", "tensor", "gpu"]
|
|
categories = ["science", "mathematics", "algorithms"]
|
|
|
|
[features]
|
|
default = ["core", "training", "inference"]
|
|
|
|
# Core (always available with feature)
|
|
core = []
|
|
|
|
# Major feature groups
|
|
training = ["rtx-transformers", "rtx-distributed", "rtx-flash-attention"]
|
|
inference = ["rtx-inference", "rtx-serving-api", "rtx-streaming"]
|
|
vision = ["rtx-vision"]
|
|
|
|
# Everything (for power users)
|
|
full = [
|
|
"core", "training", "inference", "vision",
|
|
]
|
|
|
|
[dependencies]
|
|
# Core dependencies (always included)
|
|
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" }
|
|
rtx-memory = { path = "../../core/rtx-memory", version = "1.0.0" }
|
|
|
|
# Optional dependencies (feature-gated)
|
|
rtx-transformers = { path = "../../training/rtx-transformers", version = "1.0.0", optional = true }
|
|
rtx-distributed = { path = "../../training/rtx-distributed", version = "1.0.0", optional = true }
|
|
rtx-flash-attention = { path = "../../training/rtx-flash-attention", version = "1.0.0", optional = true }
|
|
|
|
rtx-inference = { path = "../../production/rtx-inference", version = "1.0.0", optional = true }
|
|
rtx-serving-api = { path = "../../production/rtx-serving-api", version = "1.0.0", optional = true }
|
|
rtx-streaming = { path = "../../production/rtx-streaming", version = "1.0.0", optional = true }
|
|
|
|
rtx-vision = { path = "../../models/rtx-vision", version = "1.0.0", optional = true }
|
|
|
|
[dev-dependencies]
|
|
tokio-test = "0.4"
|
|
criterion = "0.5"
|
|
[lints]
|
|
workspace = true
|