Burn provides a flexible, efficient, and portable tensor library for building deep learning applications.
dfdx
Deep learning in Rust, with shape checked tensors and neural networks.

About dfdx
dfdx is a deep learning library written in Rust. It focuses on tensor shapes and types that are checked at compile time, so that some shape mismatches are caught before the code runs. The library provides tensors with GPU acceleration, a set of tensor operations, neural network building blocks, and optimizers. The project is in a pre-alpha state, and the maintainers have said the next few releases are planned to be breaking releases. The source lists version 0.13.0 as available on crates.io.
Key features
- GPU-accelerated tensor library with shapes up to six dimensions
- Tensor shapes with both compile-time and runtime-sized dimensions
- Tensor operations checked for shape and type at compile time
- Neural network building blocks such as Linear, Conv2D, and Transformer
- Optimizers including Sgd, Adam, AdamW, and RMSprop
- Optional CUDA device support enabled through a cargo feature flag
- States compile-time checking of shapes and types as a design goal
- Lists ergonomics and performance among its stated design goals
- Reports that the only current unsafe calls are for matrix multiplication
- The project is in a pre-alpha state, and the next few releases are planned to be breaking
- CUDA support requires installing NVIDIA's CUDA toolkit
- Contains unsafe code for matrix multiplication
dfdx at a glance
- Website
- github.com/coreylowman/dfdx
- Type
- Open-source project
- Pricing
- Not confirmed. Check the site
- Open source
- Not stated
- Language
- English
- Last checked
- Oct 6, 2026 · Online when checked
- Listed
- Oct 5, 2026
dfdx alternatives
Tools and sites you could use instead, ranked by how closely they match.
Tensors and Dynamic neural networks in Python with strong GPU acceleration.
Build machine learning applications with Delta, an open-source framework implemented in Rust.
Catniff provides JavaScript developers with a Torch-like deep learning framework.
Caffe provides an open framework for building and running deep learning models.
The Deep Diamond library provides fast tensor and neural network computations in Clojure using native CPU and GPU routines.
Popular in AI Infrastructure & MLOps
Well-known listings from the same topic.
dfdx FAQ
How do I add dfdx to a Rust project?
What is required to use GPU acceleration with CUDA?
Is the API stable?
Tell us and an editor will check it.
Claim the listing and suggest updates yourself. An editor checks every claim first.












