38 lines
893 B
TOML
38 lines
893 B
TOML
[package]
|
|
name = "rtx-interpret"
|
|
version = "0.1.0"
|
|
edition = "2024"
|
|
rust-version = "1.92"
|
|
authors = ["RustyTorch++ Contributors"]
|
|
description = "Model interpretability library for RustyTorch++ (Captum equivalent)"
|
|
license = "MIT OR Apache-2.0"
|
|
repository = "https://github.com/rustytorch/rustytorch"
|
|
keywords = ["machine-learning", "interpretability", "attribution", "saliency"]
|
|
categories = ["science", "algorithms"]
|
|
|
|
[dependencies]
|
|
rtx-tensor = { path = "../rtx-tensor" }
|
|
rtx-autograd = { path = "../rtx-autograd" }
|
|
rtx-nn = { path = "../rtx-nn" }
|
|
thiserror = "2.0"
|
|
tracing = "0.1"
|
|
anyhow = "1.0"
|
|
serde = { version = "1.0", features = ["derive"] }
|
|
serde_json = "1.0"
|
|
bincode = "1.3"
|
|
|
|
[dev-dependencies]
|
|
approx = "0.5"
|
|
|
|
[features]
|
|
default = []
|
|
cuda = ["rtx-tensor/cuda"]
|
|
metal = ["rtx-tensor/metal", "rtx-nn/metal"]
|
|
|
|
[lib]
|
|
name = "rtx_interpret"
|
|
path = "src/lib.rs"
|
|
|
|
[lints]
|
|
workspace = true
|