Open-source AI Library for Rust
v1.2
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.
- 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