← all repositories

chelsea0x3b/dfdx

A shape-checked deep learning library for Rust with GPU acceleration and automatic differentiation.

1.9k stars Rust ML Frameworks
dfdx
Velocity · 7d
+1.1
★ / day
Trend
steady
star history

dfdx is a Rust-based deep learning framework providing GPU-accelerated tensor operations with compile-time shape checking. It includes neural network building blocks like Linear, Conv2D, and Transformer layers, along with standard optimizers (SGD, Adam, AdamW, RMSprop). The library uses reverse-mode automatic differentiation for backpropagation and supports CUDA/cuDNN acceleration.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.