57 lines
1.6 KiB
TOML
57 lines
1.6 KiB
TOML
[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
|