cksac / rai
RAI: Rust ML framework with composable transformations like JAX.
☆88Updated 9 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☆92Updated 8 months ago
- Enzyme integration into Rust. Experimental, do not use.☆101Updated 2 years ago
- ☆85Updated 4 months ago
- A rust fork to work towards Enzyme integration☆86Updated this week
- Sample Python extension using Rust/PyO3/tch to interact with PyTorch☆35Updated last year
- Rustic bindings to the IREE Compiler/Runtime☆19Updated 3 weeks ago
- A Rust Library for High-Performance Tensor Exchange with Python☆41Updated this week
- Example of tch-rs on M1☆53Updated last year
- LLaMa 7b with CUDA acceleration implemented in rust. Minimal GPU memory needed!☆104Updated last year
- OpenAI's Gym written in pure Rust for blazingly fast performance☆97Updated 6 months ago
- Simplistic API for deep learning tensor operations☆28Updated last week
- Scientific Computing in Rust☆189Updated last year
- An extension library to Candle that provides PyTorch functions not currently available in Candle☆39Updated last year
- A rust reinforcement learning library☆34Updated 9 months ago
- A Rust implementation of Gaussian Process regression.☆56Updated last year
- Automatic differentiation in Rust with WGPU support☆23Updated 3 years ago
- ☆65Updated last year
- ☆29Updated 5 months ago
- Andrej Karpathy's Let's build GPT: from scratch video & notebook implemented in Rust + candle☆72Updated last year
- Einstein summation for Rust☆36Updated 4 years ago
- Auto differentiation library☆69Updated 2 weeks ago
- A minimal OpenCL, CUDA, Vulkan and host CPU array manipulation engine / framework.☆73Updated last month
- Generic, parallel Monte Carlo tree search library☆72Updated 2 years ago
- Efficient global optimization toolbox in Rust: bayesian optimization, mixture of gaussian processes, sampling methods☆104Updated this week
- ☆23Updated 3 weeks ago
- A collection of optimisers for use with candle☆34Updated last week
- General matrix multiplication of f32 and f64 matrices in Rust. Supports matrices with general strides.☆232Updated 9 months ago
- Low rank adaptation (LoRA) for Candle.☆145Updated 3 weeks ago
- Statistical routines for ndarray☆209Updated 8 months ago
- ☆26Updated last year