chelsea0x3b/dfdx

Health v2 · maintenance only

Flags: no_license

Adoption not part of the score

1932 stars · 107 forks observed · 2026-08-28

What it is AI-extracted, prompt v1, taxonomy v1, 2026-08-30, confidence not recorded

A Rust deep learning library providing shape-checked tensors and neural network building blocks with compile-time verification of tensor operations. It includes automatic differentiation (backpropagation), standard optimizers like SGD and Adam, and optional CUDA GPU acceleration.

Use cases

• build neural networks in rust
• train deep learning models with compile-time shape checking
• gpu accelerated tensor operations in rust
• automatic differentiation and backpropagation library for rust
• implement transformers and convnets without python
• type-safe tensor math with checked dimensions

When to choose

• you want deep learning in pure rust with strong compile-time guarantees
• you need shape errors caught at compile time instead of runtime
• you want to avoid python dependencies in your ML pipeline
• you need cuda/cudnn gpu acceleration from rust

When to avoid

• you need production stability - the project is pre-alpha with breaking releases planned
• you require the breadth and ecosystem of pytorch or tensorflow
• you need distributed training or large-scale model support
• you want extensive pretrained models and community tooling

Facets

library · maturity experimental

machine-learning deep-learning gpu-computing deep-learning machine-learning gpu-computing rust tensor-library autograd neural-networks compile-time-shape-checking backpropagation cudnn type-safe-tensors gpu cuda

Member repositories

For agents

markdown · JSON · MCP: product_card(name="chelsea0x3b/dfdx")