While native Rust deep learning libraries have made progress, they are still largely experimental. However, there are high-quality bindings available for both Tensorflow and PyTorch.
Neural Networks
tch [ crate · repo · docs ]
Rust wrappers for the PyTorch C++ api (libtorch).
tensorflow [ crate · repo · docs ]
Rust language bindings for TensorFlow.
tflite [ crate · repo · docs ]
Rust bindings for TensorFlow Lite
dfdx [ crate · repo · docs ]
Ergonomic auto differentiation in Rust, with pytorch like apis.
candle-core [ crate · repo · docs ]
Minimalist ML framework with CUDA acceleration and quantization support
tract [ crate · repo · docs ]
Tiny, no-nonsense, self contained, TensorFlow and ONNX inference
burn [ crate · repo · docs ]
Flexible and Comprehensive Deep Learning Framework in Rust
wonnx [ crate · repo · docs ]
Wonnx is an ONNX runtime based on wgpu aimed at being a universal GPU runtime, written in Rust.
neuroflow [ crate · repo · docs ]
The neural network library implemented in Rust
rusty-machine [ crate · repo · docs ]
A machine learning library.
autograd [ crate · repo · docs ]
Tensors and differentiable operations in Rust
autograph [ crate · repo · docs ]
A machine learning library for Rust.
juice [ crate · repo · docs ]
Machine Learning Framework for Hackers
cogent [ crate · repo · docs ]
Basic neural network library for classification.
tvm [ crate · repo · docs ]
Rust frontend support for TVM
nn [ crate · repo · docs ]
A multilayer feedforward backpropagation neural network library
neuronika [ crate · repo · docs ]
Tensors and dynamic neural networks.
alumina [ crate · repo · docs ]
An Experimental Deep Learning Library
auto-diff [ crate · repo · docs ]
A neural network library in Rust.
rustml [ crate · repo · docs ]
A library for doing maching learning in Rust.
leaf [ crate · repo · docs ]
Machine Learning Framework for Hackers
prophet [ crate · repo · docs ]
A neural network implementation with a focus on cache-efficiency and sequential performance.
fann [ crate · repo · docs ]
Wrapper for the Fast Artificial Neural Networks library
cntk [ crate · repo · docs ]
Wrapper around Microsoft CNTK library
revonet [ crate · repo · docs ]
Rust implementation of real-coded genetic algorithm for solving optimization problems and training of neural networks. The latter is also known as neuroevolution.
hal-ml [ crate · repo · docs ]
HAL: a machine learning library that is able to run on Nvidia, OpenCL or CPU BLAS based compute backends. It currently provides stackable classical neural networks, RNN's and soon to be LSTM's. A differentiation of this package is that we are looking to implement RTRL (instead of just BPTT) for the recurrent layers in order to provide a solid framework for online learning. We will also (in the future) be implementing various layers such as unitary RNN's, NTM's and Adaptive Computation time based LSTM's. HAL also comes with the ability to plot and do many basic math operations on arrays.
drug [ crate · repo · docs ]
A differentiable computation graph for neural networks.
Looking for something you didn't find? Try asking on Zulip chat.