[package]
name = "rtx-nlg"
version = "0.1.0"
edition = "2021"
description = "Advanced Natural Language Generation for RTX - Production-ready text generation with state-of-the-art algorithms"
license = "MIT OR Apache-2.0"
keywords = ["nlp", "nlg", "text-generation", "transformers", "machine-learning"]
categories = ["science", "text-processing"]
authors = ["RTX Team"]

[dependencies]
rtx-tensor = { path = "../rtx-tensor" }
rtx-transformers = { path = "../rtx-transformers" }
serde = { version = "1.0", features = ["derive"] }
serde_json = "1.0"
thiserror = "1.0"
anyhow = "1.0"
rayon = "1.8"
tokio = { version = "1.0", features = ["full"] }
futures = "0.3"
dashmap = "5.5"
indexmap = "2.0"
regex = "1.10"
unicode-segmentation = "1.10"
rand = "0.8"
rand_distr = "0.4"
ndarray = "0.15"
candle-core = "0.3"
candle-nn = "0.3"
tch = "0.15"
tracing = "0.1"
tracing-subscriber = "0.3"
metrics = "0.22"
lru = "0.12"
num_cpus = "1.16"
chrono = { version = "0.4", features = ["serde"] }
async-trait = "0.1"
tokio-stream = "0.1"

[dev-dependencies]
criterion = { version = "0.5", features = ["html_reports"] }
proptest = "1.4"
tempfile = "3.8"
env_logger = "0.10"

[[bench]]
name = "generation_benchmarks"
harness = false

[[bench]]
name = "translation_benchmarks"
harness = false

[[bench]]
name = "summarization_benchmarks"
harness = false

[lib]
name = "rtx_nlg"
path = "src/lib.rs"

[features]
default = ["production", "cuda"]
production = []
cuda = []
distributed = []
streaming = []