cksac / raiLinks
RAI: Rust ML framework with composable transformations like JAX.
☆94Updated last year
Alternatives and similar repositories for rai
Users that are interested in rai are comparing it to the libraries listed below
Sorting:
- Experimentation using the xla compiler from rust☆97Updated last year
- Enzyme integration into Rust. Experimental, do not use.☆101Updated 2 years ago
- ☆93Updated 9 months ago
- Sample Python extension using Rust/PyO3/tch to interact with PyTorch☆37Updated last year
- A rust fork to work towards Enzyme integration☆92Updated 3 months ago
- A Rust implementation of Gaussian Process regression.☆56Updated last year
- OpenAI's Gym written in pure Rust for blazingly fast performance☆114Updated 11 months ago
- Rustic bindings to the IREE Compiler/Runtime☆23Updated last month
- General matrix multiplication of f32 and f64 matrices in Rust. Supports matrices with general strides.☆239Updated 4 months ago
- A Deep Learning and preprocessing framework in Rust with support for CPU and GPU.☆132Updated 2 years ago
- A Rust implementation of the CMA-ES optimization algorithm.☆35Updated 4 months ago
- Efficient global optimization toolbox in Rust: bayesian optimization, mixture of gaussian processes, sampling methods☆140Updated this week
- Rust Scientific Libary. ODE and DAE (Runge-Kutta) solvers. Special functions (Bessel, Elliptic, Beta, Gamma, Erf). Linear algebra. Sparse…☆172Updated this week
- Source of BLAS and LAPACK via OpenBLAS☆89Updated last month
- Generic, parallel Monte Carlo tree search library☆73Updated 3 years ago
- A fun, hackable, GPU-accelerated, neural network library in Rust, written by an idiot☆133Updated last year
- A minimal OpenCL, CUDA, Vulkan and host CPU array manipulation engine / framework.☆75Updated last month
- Statistics, Information Measures, Linear Algebra, Cholesky Matrix Decomposition, Mahalanobis Distance, Householder QR Decomposition, Clif…☆56Updated last year
- An extension library to Candle that provides PyTorch functions not currently available in Candle☆40Updated last year
- Simplistic API for deep learning tensor operations☆30Updated 5 months ago
- A Rust Library for High-Performance Tensor Exchange with Python☆47Updated last month
- A machine learning library for Rust.☆329Updated last year
- Statistical routines for ndarray☆213Updated last year
- Automatic differentiation in Rust with WGPU support☆24Updated 4 years ago
- A rust reinforcement learning library☆43Updated last year
- BLAS source of choice☆43Updated this week
- Kiddo☆133Updated this week
- Auto differentiation library☆70Updated 5 months ago
- OpenAI Gym bindings for Rust☆69Updated 2 years ago
- ☆24Updated 5 months ago