[package]
name = "rtx-nn"
version = "1.0.0"
edition = "2024"
rust-version = "1.92"
authors = ["RustyTorch Team"]
description = "Neural network layers and operations for RustyTorch"
license = "MIT OR Apache-2.0"
repository = "https://github.com/rusty-torch/rusty-torch"
keywords = ["neural-networks", "deep-learning", "gpu", "cuda", "ai"]
categories = ["science", "mathematics"]

[dependencies]
rtx-tensor = { path = "../rtx-tensor" }
rtx-backend = { path = "../rtx-backend" }
rtx-backend-cpu = { path = "../rtx-backend-cpu" }
rtx-macros = { path = "../rtx-macros" }
cudarc = { workspace = true, optional = true }
fastrand = "2.0"
serde = { version = "1.0", features = ["derive"] }
thiserror = "2.0"
tracing = "0.1"
rand = "0.8"
rand_distr = "0.4"
num-traits = "0.2"
half = { version = "2.4", optional = true }

[target.'cfg(target_os = "macos")'.dependencies]
# Metal GPU acceleration for Apple Silicon
objc2 = { version = "0.6", optional = true, features = ["std"] }
objc2-metal = { version = "0.3", optional = true }
objc2-foundation = { version = "0.3", optional = true }
block2 = { version = "0.6", optional = true }

[dev-dependencies]
approx = "0.5"
criterion = { version = "0.5", features = ["html_reports"] }

[features]
default = []
cuda = ["rtx-tensor/cuda", "cudarc"]
metal = ["rtx-tensor/metal", "dep:objc2", "dep:objc2-metal", "dep:objc2-foundation", "dep:block2"]
f16 = ["half"]

[[example]]
name = "generic_layers_demo"
path = "examples/generic_layers_demo.rs"

[[example]]
name = "derive_macros_demo"
path = "examples/derive_macros_demo.rs"

[package.metadata.docs.rs]
features = ["cuda", "f16"]
[lints]
workspace = true