[package]
edition = "2024"
name = "burn-tensor"
version = "0.20.1"
authors = ["nathanielsimard <nathaniel.simard.42@gmail.com>"]
build = false
autolib = false
autobins = false
autoexamples = false
autotests = false
autobenches = false
description = "Tensor library with user-friendly APIs and automatic differentiation support"
documentation = "https://docs.rs/burn-tensor"
readme = "README.md"
keywords = [
"deep-learning",
"machine-learning",
"tensor",
"pytorch",
"ndarray",
]
categories = [
"science",
"no-std",
"embedded",
"wasm",
]
license = "MIT OR Apache-2.0"
repository = "https://github.com/tracel-ai/burn/tree/main/crates/burn-tensor"
resolver = "2"
[package.metadata.docs.rs]
features = ["doc"]
rustdoc-args = [
"--cfg",
"docsrs",
"--html-in-header",
"katex-header.html",
]
[features]
cubecl = [
"burn-std/cubecl",
"burn-backend/cubecl",
]
cubecl-cpu = ["burn-backend/cubecl-cpu"]
cubecl-cuda = ["burn-backend/cubecl-cuda"]
cubecl-hip = ["burn-backend/cubecl-hip"]
cubecl-wgpu = ["burn-backend/cubecl-wgpu"]
default = ["std"]
doc = ["default"]
experimental-named-tensor = []
std = [
"num-traits/std",
"burn-std/std",
"burn-backend/std",
"colored",
]
tracing = [
"burn-std/tracing",
"burn-backend/tracing",
]
[lib]
name = "burn_tensor"
path = "src/lib.rs"
[dependencies.burn-backend]
version = "0.20.1"
default-features = false
[dependencies.burn-std]
version = "0.20.1"
default-features = false
[dependencies.colored]
version = "3.0.0"
optional = true
[dependencies.derive-new]
version = "0.7.0"
default-features = false
[dependencies.num-traits]
version = "0.2.19"
features = ["libm"]
default-features = false
[dependencies.serde]
version = "1.0.228"
features = [
"derive",
"alloc",
]
default-features = false
[dev-dependencies.paste]
version = "1"
[lints.clippy]
[lints.rustdoc]
broken_intra_doc_links = "deny"
invalid_html_tags = "deny"