LaurentMazare / xla-rs
Experimentation using the xla compiler from rust
☆90Updated 7 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.☆85Updated 7 months ago
- ☆83Updated 2 months ago
- Sample Python extension using Rust/PyO3/tch to interact with PyTorch☆34Updated last year
- A Rust Library for High-Performance Tensor Exchange with Python☆40Updated 3 months ago
- Enzyme integration into Rust. Experimental, do not use.☆102Updated 2 years ago
- A rust fork to work towards Enzyme integration☆83Updated this week
- Auto differentiation library☆68Updated last year
- Read and write tensorboard data using Rust☆19Updated last year
- ☆21Updated 8 months ago
- Einstein summation for Rust☆35Updated 3 years ago
- Automatic differentiation in Rust with WGPU support☆23Updated 3 years ago
- BLAS source of choice☆40Updated last month
- A minimal OpenCL, CUDA, Vulkan and host CPU array manipulation engine / framework.☆74Updated 3 months ago
- ☆25Updated last year
- General matrix multiplication of f32 and f64 matrices in Rust. Supports matrices with general strides.☆229Updated 7 months ago
- Source of BLAS and LAPACK via OpenBLAS☆83Updated 2 months ago
- ☆28Updated 3 months ago
- Generic Automatic Differentiation library for Rust (aka "autograd")☆37Updated 3 years ago
- ☆19Updated 5 months ago
- An extension library to Candle that provides PyTorch functions not currently available in Candle☆38Updated last year
- Simplistic API for deep learning tensor operations☆28Updated last month
- .npy and .npz file format support for ndarray☆59Updated 6 months ago
- A Rust implementation of Gaussian Process regression.☆57Updated last year
- OpenAI's Gym written in pure Rust for blazingly fast performance☆94Updated 4 months ago
- Low effort scraping Python's pickle format in Rust. It is to complete pickle parsing as BeautifulSoup was to complete HTML parsing.☆25Updated 9 months ago
- A collection of optimisers for use with candle☆34Updated 3 months ago
- 🧮 alphatensor matrix breakthrough algorithms + simd + rust.☆59Updated 2 years ago
- A rust implementation of AlphaZero algorithm☆47Updated 2 years ago
- Generalized (hyper-) dual numbers in rust☆64Updated 3 months ago
- Rust library for running TensorRT accelerated deep learning models☆52Updated 3 years ago