cksac / rai
RAI: Rust ML framework with composable transformations like JAX.
☆78Updated 4 months ago
Related projects ⓘ
Alternatives and complementary repositories for rai
- Experimentation using the xla compiler from rust☆89Updated 3 months ago
- Enzyme integration into Rust. Experimental, do not use.☆102Updated last year
- A rust fork to work towards Enzyme integration☆70Updated this week
- Low rank adaptation (LoRA) for Candle.☆127Updated 3 months ago
- OpenAI's Gym written in pure Rust for blazingly fast performance☆77Updated last month
- Rustic bindings to the IREE Compiler/Runtime☆16Updated 4 months ago
- ☆76Updated 5 months ago
- Statistics, Information Measures, Linear Algebra, Cholesky Matrix Decomposition, Mahalanobis Distance, Householder QR Decomposition, Clif…☆51Updated 4 months ago
- Sample Python extension using Rust/PyO3/tch to interact with PyTorch☆32Updated 9 months ago
- Generic, parallel Monte Carlo tree search library☆65Updated 2 years ago
- Simplistic API for deep learning tensor operations☆28Updated 3 weeks ago
- Statistical routines for ndarray☆201Updated 2 months ago
- A Deep Learning and preprocessing framework in Rust with support for CPU and GPU.☆124Updated last year
- Example of tch-rs on M1☆51Updated 8 months ago
- Auto differentiation library☆62Updated last year
- Computational graphs with reverse automatic differentation in the GPU☆387Updated 2 years ago
- Rank Polymorphic array library for Rust.☆103Updated last year
- An extension library to Candle that provides PyTorch functions not currently available in Candle☆37Updated 8 months ago
- A machine learning library for Rust.☆314Updated 3 months ago
- ☆22Updated this week
- LLaMa 7b with CUDA acceleration implemented in rust. Minimal GPU memory needed!☆101Updated last year
- A Probabilistic Programming Language In Rust☆35Updated 2 years ago
- ☆19Updated 4 months ago
- A fun, hackable, GPU-accelerated, neural network library in Rust, written by an idiot☆122Updated 5 months ago
- Rust Scientific Libary. ODE and DAE (Runge-Kutta) solvers. Special functions (Bessel, Elliptic, Beta, Gamma, Erf). Linear algebra. Sparse…☆114Updated last month
- A minimal OpenCL, CUDA, Vulkan and host CPU array manipulation engine / framework.☆70Updated this week
- BLAS source of choice☆39Updated 2 months ago
- Automatic differentiation in Rust with WGPU support☆23Updated 3 years ago
- OpenAI Gym bindings for Rust☆61Updated 2 years ago
- A rust implementation of AlphaZero algorithm☆43Updated last year