chelsea0x3b/dfdx
Deep learning in Rust, with shape checked tensors and neural networks
This project helps deep learning practitioners build and train neural networks with greater confidence. You define your network architecture and training process, and the system ensures that the shapes and types of your data tensors are correct before your code even runs. It takes your raw data and network definition, processes it using GPU acceleration, and outputs trained models for tasks like classification or prediction.
1,896 stars. No commits in the last 6 months.
Use this if you are a deep learning engineer or researcher who values compile-time safety and performance in your neural network development, especially when working with Rust.
Not ideal if you are looking for a high-level, production-ready deep learning framework with extensive community support and pre-built models, as this is currently in a pre-alpha state.
Stars
1,896
Forks
104
Language
Rust
License
—
Category
Last pushed
Jul 23, 2024
Commits (30d)
0
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/chelsea0x3b/dfdx"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Higher-rated alternatives
EnzymeAD/Enzyme
High-performance automatic differentiation of LLVM and MLIR.
Oxen-AI/Oxen
Lightning fast data version control system for structured and unstructured machine learning...
LaurentMazare/tch-rs
Rust bindings for the C++ api of PyTorch.
SunDoge/dlpark
A Rust Library for High-Performance Tensor Exchange with Python
TheMesocarp/koho
Full spectrum sheaf neural network over arbitrary CW complexes.