cksac / raiLinks
RAI: Rust ML framework with composable transformations like JAX.
☆95Updated 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☆102Updated last year
- Enzyme integration into Rust. Experimental, do not use.☆102Updated 2 years ago
- ☆97Updated 2 months ago
- Sample Python extension using Rust/PyO3/tch to interact with PyTorch☆41Updated 2 years ago
- A rust fork to work towards Enzyme integration☆92Updated 7 months ago
- OpenAI's Gym written in pure Rust for blazingly fast performance☆124Updated last year
- Rustic bindings to the IREE Compiler/Runtime☆24Updated 5 months ago
- A Rust implementation of Gaussian Process regression.☆57Updated 2 years ago
- A machine learning library for Rust.☆334Updated last year
- General matrix multiplication of f32 and f64 matrices in Rust. Supports matrices with general strides.☆249Updated last month
- Experimental compiler for deep learning models☆75Updated 4 months ago
- Rust Scientific Libary. ODE and DAE (Runge-Kutta) solvers. Special functions (Bessel, Elliptic, Beta, Gamma, Erf). Linear algebra. Sparse …☆178Updated this week
- A Rust Library for High-Performance Tensor Exchange with Python☆51Updated last week
- A minimal OpenCL, CUDA, Vulkan and host CPU array manipulation engine / framework.☆77Updated 5 months ago
- Generic, parallel Monte Carlo tree search library☆77Updated 3 years ago
- Simplistic API for deep learning tensor operations☆35Updated 2 weeks ago
- Source of BLAS and LAPACK via OpenBLAS☆89Updated 3 weeks ago
- An extension library to Candle that provides PyTorch functions not currently available in Candle☆41Updated last year
- Scientific Computing in Rust☆190Updated 4 months ago
- BLAS source of choice☆45Updated 4 months ago
- A Deep Learning and preprocessing framework in Rust with support for CPU and GPU.☆133Updated 2 years ago
- Statistical routines for ndarray☆221Updated last month
- Auto differentiation library☆74Updated 9 months ago
- Tensors and differentiable operations (like TensorFlow) in Rust☆500Updated 2 years ago
- A fun, hackable, GPU-accelerated, neural network library in Rust, written by an idiot☆137Updated last year
- A lightweight gradient boosted decision tree package.☆81Updated 9 months ago
- Einstein summation for Rust☆40Updated 4 years ago
- ☆28Updated 2 years ago
- OpenAI Gym bindings for Rust☆70Updated 3 years ago
- Safe, portable, high performance compute (GPGPU) kernels.☆252Updated last month