[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