Expand description
This crate is a Rust-native tensor and math library, developed for Project Elara. It aims to be a building block of a comprehensive machine learing and scientific computing library in Rust. It tries to give a Rust experience similar to SciPy, NumPy, and PyTorch/TensorFlow, offering:
- Tensors: N-dimensional differentiable arrays (via
Tensor) - An implementation of reverse-mode autodifferentiation
- Numerical solvers for calculus (only numerical integral evaluation is fully-supported at the moment, the ODE solver has been moved to
elara-array):
Modules§
- prelude
elara-mathprelude
Macros§
- count
- A macro for counting the number of args passed to it
- scalar
- Macro for quickly creating scalar tensors
- tensor
- Macro for quickly creating tensors
Structs§
- Linear
- A 2D linearly densely-connected layer
- Model
- A neural network model with a keras-inspired API
- Tensor
- A PyTorch-like differentiable tensor type
- Tensor
Data - Backing data for
Tensor
Enums§
- Activations
- Common activation functions
- Optimizers
- Common optimizers
Traits§
- Layer
- A general trait of a layer of a neural network
Functions§
- mse
- Mean squared error function