[package]
edition = "2024"
name = "etensor-core"
version = "0.0.1"
build = false
autolib = false
autobins = false
autoexamples = false
autotests = false
autobenches = false
description = "The pure Rust tensor math and autograd engine"
readme = "README.md"
license = "MIT OR Apache-2.0"
resolver = "2"
[features]
cuda-native = []
default = []
torch = []
[lib]
name = "etensor_core"
path = "src/lib.rs"
[[test]]
name = "autograd_tests"
path = "tests/autograd_tests.rs"
[[test]]
name = "core_tests"
path = "tests/core_tests.rs"
[[bench]]
name = "bench_autograd"
path = "benches/bench_autograd.rs"
harness = false
[[bench]]
name = "bench_math"
path = "benches/bench_math.rs"
harness = false
[dependencies.matrixmultiply]
version = "0.3.10"
features = ["threading"]
[dependencies.ndarray]
version = "0.17.2"
features = ["blas"]
[dependencies.rayon]
version = "1.12.0"
[dependencies.safetensors]
version = "0.7.0"
[dependencies.thiserror]
version = "2.0.18"
[dev-dependencies.criterion]
version = "0.5"