Skip to content

Latest commit

 

History

History
37 lines (30 loc) · 716 Bytes

README.md

File metadata and controls

37 lines (30 loc) · 716 Bytes

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