LaurentMazare / xla-rsLinks
Experimentation using the xla compiler from rust
☆93Updated 9 months ago
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.☆89Updated 10 months ago
- ☆86Updated 4 months ago
- Enzyme integration into Rust. Experimental, do not use.☆101Updated 2 years ago
- A Rust Library for High-Performance Tensor Exchange with Python☆45Updated this week
- Sample Python extension using Rust/PyO3/tch to interact with PyTorch☆36Updated last year
- A rust fork to work towards Enzyme integration☆89Updated last week
- Automatic differentiation in Rust with WGPU support☆23Updated 3 years ago
- ☆23Updated last month
- HowTo: Compile CUDA with nvcc, and link to Rust☆49Updated 6 years ago
- Einstein summation for Rust☆38Updated 4 years ago
- Auto differentiation library☆69Updated last month
- BLAS source of choice☆42Updated 3 months ago
- ☆26Updated last year
- .npy and .npz file format support for ndarray☆63Updated 8 months ago
- Rust bindings to the MLIR C API.☆65Updated 2 months ago
- General matrix multiplication of f32 and f64 matrices in Rust. Supports matrices with general strides.☆238Updated 3 weeks ago
- Source of BLAS and LAPACK via OpenBLAS☆85Updated 5 months ago
- Simplistic API for deep learning tensor operations☆30Updated last month
- Voracious radix sort☆68Updated last year
- Rustic bindings to the IREE Compiler/Runtime☆20Updated 3 weeks ago
- A minimal OpenCL, CUDA, Vulkan and host CPU array manipulation engine / framework.☆73Updated last week
- An extension library to Candle that provides PyTorch functions not currently available in Candle☆39Updated last year
- Rustic bindings for IREE☆18Updated 2 years ago
- Coordination repository of the Machine Learning (applicant) Working Group☆18Updated 4 years ago
- Experimental compiler for deep learning models☆67Updated 2 weeks ago
- A Rust implementation of Gaussian Process regression.☆56Updated last year
- Redistribute Intel MKL as a crate☆89Updated last year
- Tensor library for machine learning☆27Updated 2 weeks ago
- A Reinforcement Learning / Neural Network library, written in Rust.☆20Updated 4 years ago
- An Extensible Compiler IR Framework☆174Updated 2 weeks ago