LaurentMazare / xla-rsLinks
Experimentation using the xla compiler from rust
☆99Updated last year
Alternatives and similar repositories for xla-rs
Users that are interested in xla-rs are comparing it to the libraries listed below
Sorting:
- RAI: Rust ML framework with composable transformations like JAX.☆93Updated last year
- A rust fork to work towards Enzyme integration☆91Updated 5 months ago
- ☆96Updated 3 weeks ago
- Enzyme integration into Rust. Experimental, do not use.☆102Updated 2 years ago
- Sample Python extension using Rust/PyO3/tch to interact with PyTorch☆39Updated last year
- Source of BLAS and LAPACK via OpenBLAS☆89Updated 3 months ago
- BLAS source of choice☆44Updated 2 months ago
- Einstein summation for Rust☆40Updated 4 years ago
- A Rust Library for High-Performance Tensor Exchange with Python☆48Updated 2 weeks ago
- Auto differentiation library☆71Updated 7 months ago
- General matrix multiplication of f32 and f64 matrices in Rust. Supports matrices with general strides.☆242Updated 6 months ago
- Automatic differentiation in Rust with WGPU support☆24Updated 4 years ago
- HowTo: Compile CUDA with nvcc, and link to Rust☆51Updated 7 years ago
- Wrappers for BLAS (Fortran)☆81Updated 10 months ago
- Experimental compiler for deep learning models☆71Updated 2 months ago
- A Rust implementation of Gaussian Process regression.☆57Updated last year
- Redistribute Intel MKL as a crate☆90Updated last year
- ☆28Updated 2 years ago
- Rustic bindings to the IREE Compiler/Runtime☆24Updated 3 months ago
- .npy and .npz file format support for ndarray☆67Updated last week
- Generic, parallel Monte Carlo tree search library☆75Updated 3 years ago
- A minimal OpenCL, CUDA, Vulkan and host CPU array manipulation engine / framework.☆76Updated 3 months ago
- A Rust implementation of the CMA-ES optimization algorithm.☆35Updated 6 months ago
- Rustic bindings for IREE☆18Updated 2 years ago
- An extension library to Candle that provides PyTorch functions not currently available in Candle☆40Updated last year
- N-dimensional interpolation methods in Rust and Python, no-std compatible☆42Updated 3 weeks ago
- A rust implementation of AlphaZero algorithm☆53Updated 2 years ago
- Dask scheduler written in Rust☆108Updated last year
- LLaMa 7b with CUDA acceleration implemented in rust. Minimal GPU memory needed!☆110Updated 2 years ago
- Rust binding to CUDA APIs☆118Updated 8 months ago