[[bench]]
harness = false
name = "batchnorm2d"
[[bench]]
harness = false
name = "conv2d"
[[bench]]
harness = false
name = "sum"
[[bench]]
harness = false
name = "softmax"
[build-dependencies.glob]
optional = true
version = "0.3.1"
[dependencies.cblas-sys]
default-features = false
optional = true
version = "0.1.4"
[dependencies.cudarc]
default-features = false
features = ["driver", "cublas", "nvrtc"]
optional = true
version = "0.9.7"
[dependencies.libc]
default-features = false
optional = true
version = "0.2"
[dependencies.matrixmultiply]
default-features = false
optional = true
version = "0.3.2"
[dependencies.memmap2]
default-features = false
optional = true
version = "0.5"
[dependencies.no-std-compat]
default-features = false
features = ["alloc", "compat_hash"]
optional = true
version = "0.4.1"
[dependencies.num-traits]
default-features = false
version = "0.2.15"
[dependencies.rand]
default-features = false
features = ["std_rng"]
version = "0.8.5"
[dependencies.rand_distr]
default-features = false
version = "0.4.3"
[dependencies.safetensors]
default-features = false
optional = true
version = "0.3"
[dependencies.spin]
default-features = false
features = ["spin_mutex", "rwlock", "portable_atomic"]
optional = true
version = "0.9.8"
[dependencies.zip]
default-features = false
optional = true
version = "0.6.2"
[dev-dependencies.indicatif]
version = "0.17.3"
[dev-dependencies.mnist]
version = "0.5.0"
[dev-dependencies.tempfile]
version = "3.3.0"
[features]
ci-check = ["cudarc?/ci-check"]
cpu-mkl-matmul = ["dep:cblas-sys", "dep:libc"]
cpu-par-matmul = ["std", "dep:matrixmultiply", "matrixmultiply?/threading"]
cpu-seq-matmul = ["dep:matrixmultiply"]
cuda = ["dep:cudarc", "dep:glob"]
cudnn = ["cuda", "cudarc?/cudnn"]
default = ["std", "fast-alloc", "cpu-par-matmul"]
fast-alloc = ["std"]
nightly = []
no-std = ["no-std-compat", "dep:spin", "cudarc?/no-std", "num-traits/libm"]
numpy = ["dep:zip", "std"]
safetensors = ["dep:safetensors", "std", "dep:memmap2"]
std = ["cudarc?/std", "matrixmultiply?/std", "rand_distr/std_math"]
test-f64 = []
test-integrations = []
[package]
description = "Ergonomic auto differentiation in Rust, with pytorch like apis."
documentation = "https://docs.rs/dfdx"
edition = "2021"
homepage = "https://github.com/coreylowman/dfdx"
keywords = ["deep-learning", "neural-network", "backprop", "tensor", "autodiff"]
license = "MIT OR Apache-2.0"
name = "dfdx"
readme = "README.md"
repository = "https://github.com/coreylowman/dfdx"
rust-version = "1.65"
version = "0.11.2"
[package.metadata.docs.rs]
features = ["nightly", "numpy", "safetensors", "cuda", "ci-check"]