53 lines
1.1 KiB
TOML
53 lines
1.1 KiB
TOML
[package]
|
|
name = "rtx-burn"
|
|
version.workspace = true
|
|
edition.workspace = true
|
|
authors.workspace = true
|
|
license.workspace = true
|
|
repository.workspace = true
|
|
description = "Burn ML framework integration for RustyTorch++ - multi-backend inference and training"
|
|
|
|
[dependencies]
|
|
# Burn ML framework with multiple backends
|
|
burn.workspace = true
|
|
burn-tensor.workspace = true
|
|
burn-ndarray = { workspace = true, optional = true }
|
|
burn-wgpu = { workspace = true, optional = true }
|
|
|
|
# Internal RTX crates
|
|
rtx-tensor = { workspace = true, features = ["cpu"] }
|
|
rtx-runtime = { workspace = true }
|
|
|
|
# Tensor data conversion
|
|
ndarray.workspace = true
|
|
|
|
# Error handling
|
|
thiserror.workspace = true
|
|
anyhow.workspace = true
|
|
|
|
# Async support
|
|
tokio.workspace = true
|
|
async-trait.workspace = true
|
|
|
|
# Serialization
|
|
serde.workspace = true
|
|
|
|
# Logging
|
|
tracing.workspace = true
|
|
|
|
[dev-dependencies]
|
|
tokio-test.workspace = true
|
|
tempfile.workspace = true
|
|
|
|
[features]
|
|
default = ["ndarray"]
|
|
# CPU backend via ndarray
|
|
ndarray = ["dep:burn-ndarray"]
|
|
# WebGPU backend (cross-platform GPU)
|
|
wgpu = ["dep:burn-wgpu"]
|
|
# All backends
|
|
full = ["ndarray", "wgpu"]
|
|
|
|
[lints]
|
|
workspace = true
|