Neural Networks

ort [ crate · repo · docs ]

A safe Rust wrapper for ONNX Runtime 1.28 - Optimize and accelerate machine learning inference & training

candle-core [ crate · repo · docs ]

Minimalist ML framework with CUDA acceleration and quantization support

tch [ crate · repo · docs ]

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

burn [ crate · repo · docs ]

Flexible and Comprehensive Deep Learning Framework in Rust

tract [ crate · repo · docs ]

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

tensorflow [ crate · repo · docs ] archived

Rust language bindings for TensorFlow.

autograd [ crate · repo · docs ]

Tensors and differentiable operations in Rust

tflite [ crate · repo · docs ]

Rust bindings for TensorFlow Lite

rusty-machine [ crate · repo · docs ] archived

A machine learning library.

dfdx [ crate · repo · docs ]

Ergonomic auto differentiation in Rust, with pytorch like apis.

custos [ crate · repo · docs ]

A minimal OpenCL, WGPU, CUDA and host CPU array manipulation engine.

luminal [ crate · repo · docs ]

Deep learning at the speed of light.

zyx [ crate · repo · docs ]

Zyx machine learning library

neuroflow [ crate · repo · docs ]

The neural network library implemented in Rust

convolutions-rs [ crate · repo · docs ]

Torch-compatible convolutions crate implemented in pure rust.

wonnx [ crate · repo · docs ] archived

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

cogent [ crate · repo · docs ]

Basic neural network library for classification.

nn [ crate · repo · docs ]

A multilayer feedforward backpropagation neural network library

prophet [ crate · repo · docs ]

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

auto-diff [ crate · repo · docs ]

A neural network library in Rust.

leaf [ crate · repo · docs ]

Machine Learning Framework for Hackers

alumina [ crate · repo · docs ]

An Experimental Deep Learning Library

tvm [ crate · repo · docs ]

Rust frontend support for TVM

rustml [ crate · repo · docs ]

A library for doing maching learning in Rust.

fann [ crate · repo · docs ]

Wrapper for the Fast Artificial Neural Networks library

mininn [ crate · repo · docs ]

A minimalist deep learnig crate for rust.

autograph [ crate · repo · docs ]

A machine learning library for Rust.

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.

juice [ crate · repo · docs ]

Machine Learning Framework for Hackers

cntk [ crate · repo · docs ]

Wrapper around Microsoft CNTK library

kyanite [ crate · repo · docs ]

A modular, expandable, gallery collector.

neuronika [ crate · repo · docs ]

Tensors and dynamic neural networks.

drug [ crate · repo · docs ]

A differentiable computation graph for neural networks.

blackportal-ai/delta [ repo ]

YdrMaster/operators-rs [ repo ]

InfiniTensor/InfiniLM [ repo ]

EricLBuehler/mistral.rs [ repo ]

outbrain/fwumious_wabbit [ repo ]

tedsta/deeplearn-rs [ repo ] archived

torchrs/torchrs [ repo ]

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.