Neural Networks

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.

tch [ crate · repo · docs ]

Rust wrappers for the PyTorch C++ api (libtorch).

Last Commit: 2023-10-01 Last Published: 2023-05-18

tensorflow [ crate · repo · docs ]

Rust language bindings for TensorFlow.

Last Commit: 2023-09-12 Last Published: 2023-08-15

tflite [ crate · repo · docs ]

Rust bindings for TensorFlow Lite

Last Commit: 2023-06-26 Last Published: 2023-06-26

dfdx [ crate · repo · docs ]

Ergonomic auto differentiation in Rust, with pytorch like apis.

Last Commit: 2023-09-28 Last Published: 2023-07-27

candle-core [ crate · repo · docs ]

Minimalist ML framework with CUDA acceleration and quantization support

Last Commit: 2023-10-01 Last Published: 2023-09-18

tract [ crate · repo · docs ]

Tiny, no-nonsense, self contained, TensorFlow and ONNX inference

Last Commit: 2023-09-30 Last Published: 2023-08-30

burn [ crate · repo · docs ]

Flexible and Comprehensive Deep Learning Framework in Rust

Last Commit: 2023-10-01 Last Published: 2023-09-06

wonnx [ crate · repo · docs ]

Wonnx is an ONNX runtime based on wgpu aimed at being a universal GPU runtime, written in Rust.

Last Commit: 2023-09-30 Last Published: 2023-09-30

neuroflow [ crate · repo · docs ]

The neural network library implemented in Rust

Last Commit: 2023-08-18 Last Published: 2023-08-18

rusty-machine [ crate · repo · docs ]

A machine learning library.

Last Commit: 2020-07-10 Last Published: 2017-01-06

autograd [ crate · repo · docs ]

Tensors and differentiable operations in Rust

Last Commit: 2023-02-11 Last Published: 2021-11-28

autograph [ crate · repo · docs ]

A machine learning library for Rust.

Last Commit: 2023-09-30 Last Published: 2021-12-12

juice [ crate · repo · docs ]

Machine Learning Framework for Hackers

Last Commit: 2023-06-21 Last Published: 2021-11-21

cogent [ crate · repo · docs ]

Basic neural network library for classification.

Last Commit: 2022-06-08 Last Published: 2021-03-03

tvm [ crate · repo · docs ]

Rust frontend support for TVM

Last Commit: 2023-10-01 Last Published: 2021-02-23

nn [ crate · repo · docs ]

A multilayer feedforward backpropagation neural network library

Last Commit: 2017-12-21 Last Published: 2015-12-11

neuronika [ crate · repo · docs ]

Tensors and dynamic neural networks.

Last Commit: 2022-10-10 Last Published: 2022-01-01

alumina [ crate · repo · docs ]

An Experimental Deep Learning Library

Last Commit: 2022-10-23 Last Published: 2018-08-17

auto-diff [ crate · repo · docs ]

A neural network library in Rust.

Last Commit: 2022-09-08 Last Published: 2022-03-28

rustml [ crate · repo · docs ]

A library for doing maching learning in Rust.

Last Commit: 2022-06-15 Last Published: 2018-01-31

leaf [ crate · repo · docs ]

Machine Learning Framework for Hackers

Last Commit: 2018-01-14 Last Published: 2016-04-23

prophet [ crate · repo · docs ]

A neural network implementation with a focus on cache-efficiency and sequential performance.

Last Commit: 2019-07-31 Last Published: 2017-10-12

fann [ crate · repo · docs ]

Wrapper for the Fast Artificial Neural Networks library

Last Commit: 2018-10-19 Last Published: 2018-10-19

cntk [ crate · repo · docs ]

Wrapper around Microsoft CNTK library

Last Commit: 2018-03-04 Last Published: 2017-11-29

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.

Last Commit: 2017-08-20 Last Published: 2017-08-20

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.

Last Commit: 2016-10-22 Last Published: 2016-06-17

drug [ crate · repo · docs ]

A differentiable computation graph for neural networks.

Last Commit: 2018-10-26 Last Published: 2018-10-21

Do you know of a crate that is missing? Create an issue or a PR.
Looking for something you didn't find? Try asking on Zulip chat.