[[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.cudarc]
default-features = false
features = ["driver", "cublas", "nvrtc"]
optional = true
version = "0.9.13"
[dependencies.gemm]
default-features = false
optional = true
version = "0.15.4"
[dependencies.half]
features = ["num-traits", "rand_distr"]
optional = true
version = "2.3.1"
[dependencies.libm]
version = "0.2.7"
[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.rayon]
optional = true
version = "1.7.0"
[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.6"
[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 = ["dep:gemm", "dep:rayon"]
cuda = ["dep:cudarc", "dep:glob"]
cudnn = ["cuda", "cudarc?/cudnn"]
default = ["std", "fast-alloc", "cpu"]
f16 = ["dep:half", "cudarc?/f16"]
fast-alloc = ["std"]
nightly = ["half?/use-intrinsics", "gemm?/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", "rand_distr/std_math", "gemm?/std"]
test-amp-f16 = ["f16"]
test-f16 = ["f16"]
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.13.0"
[package.metadata.docs.rs]
features = ["nightly", "numpy", "safetensors", "cuda", "ci-check"]