cksac / rai
RAI: Rust ML framework with composable transformations like JAX.
☆85Updated 8 months ago
Alternatives and similar repositories for rai:
Users that are interested in rai are comparing it to the libraries listed below
- Experimentation using the xla compiler from rust☆90Updated 7 months ago
- Enzyme integration into Rust. Experimental, do not use.☆102Updated 2 years ago
- ☆84Updated 2 months ago
- A rust fork to work towards Enzyme integration☆83Updated this week
- Auto differentiation library☆68Updated last year
- Example of tch-rs on M1☆53Updated last year
- OpenAI's Gym written in pure Rust for blazingly fast performance☆94Updated 5 months ago
- A Rust implementation of Gaussian Process regression.☆57Updated last year
- A rust reinforcement learning library☆30Updated 7 months ago
- An extension library to Candle that provides PyTorch functions not currently available in Candle☆38Updated last year
- LLaMa 7b with CUDA acceleration implemented in rust. Minimal GPU memory needed!☆104Updated last year
- ☆66Updated last year
- Experimental compiler for deep learning models☆28Updated last month
- Sample Python extension using Rust/PyO3/tch to interact with PyTorch☆34Updated last year
- A minimal OpenCL, CUDA, Vulkan and host CPU array manipulation engine / framework.☆73Updated this week
- OpenAI Gym bindings for Rust☆63Updated 2 years ago
- Low rank adaptation (LoRA) for Candle.☆144Updated 7 months ago
- A fun, hackable, GPU-accelerated, neural network library in Rust, written by an idiot☆128Updated 9 months ago
- ☆25Updated last year
- Automatic differentiation in Rust with WGPU support☆23Updated 3 years ago
- A Rust Library for High-Performance Tensor Exchange with Python☆40Updated 4 months ago
- Generic, parallel Monte Carlo tree search library☆72Updated 2 years ago
- Rustic bindings to the IREE Compiler/Runtime☆17Updated last month
- A reinforcement learning library in Rust☆45Updated last week
- ☆28Updated 4 months ago
- Simplistic API for deep learning tensor operations☆28Updated last month
- A Rust implementation of the CMA-ES optimization algorithm.☆35Updated 3 months ago
- Source of BLAS and LAPACK via OpenBLAS☆84Updated 3 months ago
- Tensor library for machine learning☆26Updated last week
- Rank Polymorphic array library for Rust.☆102Updated last year