cksac / raiLinks
RAI: Rust ML framework with composable transformations like JAX.
☆89Updated 10 months ago
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☆93Updated 9 months ago
- Enzyme integration into Rust. Experimental, do not use.☆101Updated 2 years ago
- ☆86Updated 4 months ago
- A rust fork to work towards Enzyme integration☆89Updated last week
- Rustic bindings to the IREE Compiler/Runtime☆20Updated 3 weeks ago
- Sample Python extension using Rust/PyO3/tch to interact with PyTorch☆36Updated last year
- OpenAI's Gym written in pure Rust for blazingly fast performance☆99Updated 7 months ago
- ☆23Updated last month
- Statistics, Information Measures, Linear Algebra, Cholesky Matrix Decomposition, Mahalanobis Distance, Householder QR Decomposition, Clif…☆52Updated 10 months ago
- LLaMa 7b with CUDA acceleration implemented in rust. Minimal GPU memory needed!☆104Updated last year
- A minimal OpenCL, CUDA, Vulkan and host CPU array manipulation engine / framework.☆73Updated last week
- Automatic differentiation in Rust with WGPU support☆23Updated 3 years ago
- An extension library to Candle that provides PyTorch functions not currently available in Candle☆39Updated last year
- OpenAI Gym bindings for Rust☆66Updated 2 years ago
- A Deep Learning and preprocessing framework in Rust with support for CPU and GPU.☆131Updated last year
- Experimental compiler for deep learning models☆67Updated 2 weeks ago
- ☆30Updated 6 months ago
- Example of tch-rs on M1☆53Updated last year
- A Rust Library for High-Performance Tensor Exchange with Python☆45Updated this week
- A rust reinforcement learning library☆36Updated 10 months ago
- Safe, portable, high performance compute (GPGPU) kernels.☆229Updated last year
- Rust Scientific Libary. ODE and DAE (Runge-Kutta) solvers. Special functions (Bessel, Elliptic, Beta, Gamma, Erf). Linear algebra. Sparse…☆152Updated this week
- Auto differentiation library☆69Updated last month
- A Rust implementation of Gaussian Process regression.☆56Updated last year
- .npy and .npz file format support for ndarray☆63Updated 8 months ago
- Python tqdm in Rust.☆45Updated last year
- ☆66Updated last year
- Andrej Karpathy's Let's build GPT: from scratch video & notebook implemented in Rust + candle☆71Updated last year
- A fun, hackable, GPU-accelerated, neural network library in Rust, written by an idiot☆131Updated last year
- ☆26Updated last year