RustML

Open-source AI Library for Rust
v1.2

About

RustML is an AI/machine learning library built for the Rust programming language. It's desgined to work similarly to Python libraries, such as PyTorch & Tensorflow.

Current Functions & Features

• Sigmoid function

Sigmoid(x: f64) -> f64

• ReLU function

ReLU(x: f64) -> f64

• Softmax function

Softmax(x: &[f64]) -> f64

• RNN algoritm

RNN(x: &[f64], y: &[f64], weights: &[f64], bias: f64) -> Vec<f64> 

• LSTM algoritm

LSTM(x: &[f64], y: &[f64], weights: &[f64]) -> Vec<f64> 

• Gradient

gradient(x: f64, y: f64, loss: f64) -> f64