LaurentMazare / xla-rs
Experimentation using the xla compiler from rust
☆91Updated 8 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.☆87Updated 9 months ago
- Enzyme integration into Rust. Experimental, do not use.☆102Updated 2 years ago
- Sample Python extension using Rust/PyO3/tch to interact with PyTorch☆34Updated last year
- A rust fork to work towards Enzyme integration☆86Updated this week
- Automatic differentiation in Rust with WGPU support☆23Updated 3 years ago
- ☆85Updated 3 months ago
- A Rust Library for High-Performance Tensor Exchange with Python☆41Updated 5 months ago
- Source of BLAS and LAPACK via OpenBLAS☆84Updated 3 months ago
- Auto differentiation library☆68Updated last year
- BLAS source of choice☆41Updated 2 months ago
- General matrix multiplication of f32 and f64 matrices in Rust. Supports matrices with general strides.☆229Updated 8 months ago
- Experimental compiler for deep learning models☆35Updated last week
- Rust bindings to the MLIR C API.☆65Updated 2 weeks ago
- HowTo: Compile CUDA with nvcc, and link to Rust☆49Updated 6 years ago
- Rustic bindings for IREE☆18Updated 2 years ago
- Einstein summation for Rust☆36Updated 4 years ago
- ☆26Updated last year
- ☆23Updated this week
- ☆28Updated 5 months ago
- A minimal OpenCL, CUDA, Vulkan and host CPU array manipulation engine / framework.☆73Updated 3 weeks ago
- Rust binding to CUDA APIs☆120Updated 3 weeks ago
- Rustic bindings to the IREE Compiler/Runtime☆18Updated 3 weeks ago
- A neural network inference library, written in Rust.☆61Updated 9 months ago
- Read and write tensorboard data using Rust☆20Updated last year
- A collection of optimisers for use with candle☆34Updated 5 months ago
- Wrappers for BLAS (Fortran)☆81Updated 2 months ago
- Redistribute Intel MKL as a crate☆88Updated 11 months ago
- A rust implementation of AlphaZero algorithm☆47Updated 2 years ago
- A Rust implementation of Gaussian Process regression.☆56Updated last year
- Generic, parallel Monte Carlo tree search library☆72Updated 2 years ago