cksac / raiLinks
RAI: Rust ML framework with composable transformations like JAX.
☆90Updated 11 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☆94Updated 10 months ago
- Enzyme integration into Rust. Experimental, do not use.☆101Updated 2 years ago
- A rust fork to work towards Enzyme integration☆90Updated 3 weeks ago
- Rustic bindings to the IREE Compiler/Runtime☆20Updated last month
- ☆88Updated 5 months ago
- OpenAI's Gym written in pure Rust for blazingly fast performance☆102Updated 8 months ago
- A rust reinforcement learning library☆36Updated 10 months ago
- A Rust Library for High-Performance Tensor Exchange with Python☆47Updated 2 weeks ago
- Simplistic API for deep learning tensor operations☆30Updated last month
- ☆30Updated 7 months ago
- Sample Python extension using Rust/PyO3/tch to interact with PyTorch☆37Updated last year
- Auto differentiation library☆69Updated 2 months ago
- Efficient global optimization toolbox in Rust: bayesian optimization, mixture of gaussian processes, sampling methods☆121Updated this week
- A minimal OpenCL, CUDA, Vulkan and host CPU array manipulation engine / framework.☆74Updated 2 weeks ago
- Statistics, Information Measures, Linear Algebra, Cholesky Matrix Decomposition, Mahalanobis Distance, Householder QR Decomposition, Clif…☆53Updated 11 months ago
- Python tqdm in Rust.☆45Updated last year
- Rust Scientific Libary. ODE and DAE (Runge-Kutta) solvers. Special functions (Bessel, Elliptic, Beta, Gamma, Erf). Linear algebra. Sparse…☆159Updated last week
- Source of BLAS and LAPACK via OpenBLAS☆85Updated this week
- Example of tch-rs on M1☆53Updated last year
- LLaMa 7b with CUDA acceleration implemented in rust. Minimal GPU memory needed!☆106Updated last year
- A Deep Learning and preprocessing framework in Rust with support for CPU and GPU.☆131Updated last year
- A collection of optimisers for use with candle☆36Updated last month
- .npy and .npz file format support for ndarray☆63Updated 9 months ago
- Rustic bindings for IREE☆18Updated 2 years ago
- A Rust implementation of the CMA-ES optimization algorithm.☆35Updated last month
- BLAS source of choice☆42Updated 4 months ago
- Statistical routines for ndarray☆212Updated 10 months ago
- Low rank adaptation (LoRA) for Candle.☆150Updated 2 months ago
- Generalized (hyper-) dual numbers in rust☆69Updated last week
- General matrix multiplication of f32 and f64 matrices in Rust. Supports matrices with general strides.☆238Updated last month