LaurentMazare / xla-rsLinks
Experimentation using the xla compiler from rust
☆96Updated 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.☆94Updated last year
- Enzyme integration into Rust. Experimental, do not use.☆101Updated 2 years ago
- A rust fork to work towards Enzyme integration☆92Updated 2 months ago
- ☆92Updated 8 months ago
- Source of BLAS and LAPACK via OpenBLAS☆89Updated last week
- Sample Python extension using Rust/PyO3/tch to interact with PyTorch☆38Updated last year
- General matrix multiplication of f32 and f64 matrices in Rust. Supports matrices with general strides.☆239Updated 4 months ago
- A Rust Library for High-Performance Tensor Exchange with Python☆47Updated last week
- Automatic differentiation in Rust with WGPU support☆23Updated 3 years ago
- Einstein summation for Rust☆40Updated 4 years ago
- Auto differentiation library☆70Updated 4 months ago
- BLAS source of choice☆43Updated last month
- Generic, parallel Monte Carlo tree search library☆72Updated 3 years ago
- Wrappers for BLAS (Fortran)☆81Updated 7 months ago
- HowTo: Compile CUDA with nvcc, and link to Rust☆50Updated 7 years ago
- Rustic bindings to the IREE Compiler/Runtime☆23Updated 3 weeks ago
- A Rust implementation of Gaussian Process regression.☆56Updated last year
- A Rust implementation of the CMA-ES optimization algorithm.☆35Updated 3 months ago
- A minimal OpenCL, CUDA, Vulkan and host CPU array manipulation engine / framework.☆75Updated 3 weeks ago
- .npy and .npz file format support for ndarray☆62Updated last year
- Experimental compiler for deep learning models☆66Updated this week
- A rust implementation of AlphaZero algorithm☆51Updated 2 years ago
- Generalized (hyper-) dual numbers in rust☆71Updated this week
- ☆27Updated 2 years ago
- Redistribute Intel MKL as a crate☆89Updated last year
- Rust bindings to the MLIR C API.☆67Updated 2 weeks ago
- An Extensible Compiler IR Framework☆198Updated last week
- l2 is a fast, Pytorch-style Tensor+Autograd library written in Rust☆197Updated 2 years ago
- Rust binding to CUDA APIs☆120Updated 5 months ago
- Reverse-mode automatic differentiation in Rust (experiment)☆63Updated 4 years ago