LaurentMazare / xla-rs
Experimentation using the xla compiler from rust
☆90Updated 5 months ago
Alternatives and similar repositories for xla-rs:
Users that are interested in xla-rs are comparing it to the libraries listed below
- RAI: Rust ML framework with composable transformations like JAX.☆82Updated 6 months ago
- Sample Python extension using Rust/PyO3/tch to interact with PyTorch☆34Updated 11 months ago
- ☆79Updated 3 weeks ago
- Enzyme integration into Rust. Experimental, do not use.☆102Updated last year
- A rust fork to work towards Enzyme integration☆80Updated this week
- Read and write tensorboard data using Rust☆19Updated 11 months ago
- A Rust Library for High-Performance Tensor Exchange with Python☆39Updated 2 months ago
- Source of BLAS and LAPACK via OpenBLAS☆80Updated last month
- Automatic differentiation in Rust with WGPU support☆23Updated 3 years ago
- ☆20Updated 6 months ago
- Experimental compiler for deep learning models☆24Updated 2 weeks ago
- Auto differentiation library☆64Updated last year
- ☆25Updated last year
- Einstein summation for Rust☆34Updated 3 years ago
- An extension library to Candle that provides PyTorch functions not currently available in Candle☆37Updated 10 months ago
- ☆27Updated 2 months ago
- Rustic bindings for IREE☆18Updated last year
- .npy and .npz file format support for ndarray☆59Updated 4 months ago
- Rust bindings to the MLIR C API.☆62Updated last month
- BLAS source of choice☆40Updated 5 months ago
- A collection of optimisers for use with candle☆31Updated 2 months ago
- Generic Automatic Differentiation library for Rust (aka "autograd")☆36Updated 3 years ago
- Generalized (hyper-) dual numbers in rust☆62Updated last month
- ☆17Updated 3 months ago
- ☆27Updated this week
- A Rust implementation of Gaussian Process regression.☆56Updated last year
- LLaMa 7b with CUDA acceleration implemented in rust. Minimal GPU memory needed!☆102Updated last year
- A rust implementation of AlphaZero algorithm☆45Updated last year
- Voracious radix sort☆65Updated last year
- Generic, parallel Monte Carlo tree search library☆67Updated 2 years ago