dfdx
A deep learning library in Rust featuring shape-checked tensors and neural networks with compile-time safety.
What is dfdx?
dfdx is a deep learning library for Rust that prioritizes ergonomics and safety through compile-time shape checking. It provides a tensor library with GPU acceleration and a comprehensive set of neural network building blocks, enabling robust and performant machine learning development.
Rust developers building deep learning models who prioritize compile-time correctness, safety, and performance, particularly those working on projects where runtime shape errors are unacceptable.
Developers choose dfdx for its unique combination of ergonomic API, compile-time shape and type checking that prevents runtime errors, and performance through GPU acceleration with minimal unsafe code and dynamic borrow checks.
Overview
Deep learning in Rust, with shape checked tensors and neural networks
Use Cases
Best For
- Building neural networks in Rust with compile-time shape safety to eliminate runtime tensor dimension errors.
Related Projects
Found a gem we're missing?
Open-Awesome is built by the community, for the community. Submit a project, suggest an awesome list, or help improve the catalog on GitHub.
