LaurentMazare / xla-rs
Experimentation using the xla compiler from rust
☆90Updated last month
Related projects: ⓘ
- RAI: Rust ML framework with composable transformations like JAX.☆68Updated 2 months ago
- ☆76Updated 3 months ago
- Enzyme integration into Rust. Experimental, do not use.☆102Updated last year
- Sample Python extension using Rust/PyO3/tch to interact with PyTorch☆30Updated 7 months ago
- A rust fork to work towards Enzyme integration☆65Updated last week
- A Rust Library for High-Performance Tensor Exchange with Python☆37Updated last month
- Source of BLAS and LAPACK via OpenBLAS☆76Updated last month
- Automatic differentiation in Rust with WGPU support☆22Updated 2 years ago
- ☆22Updated 2 months ago
- Auto differentiation library☆64Updated last year
- Rustic bindings for IREE☆17Updated last year
- .npy and .npz file format support for ndarray☆54Updated this week
- Einstein summation for Rust☆34Updated 3 years ago
- General matrix multiplication of f32 and f64 matrices in Rust. Supports matrices with general strides.☆208Updated last month
- ☆25Updated last year
- An extension library to Candle that provides PyTorch functions not currently available in Candle☆35Updated 6 months ago
- Rustic bindings to the IREE Compiler/Runtime☆16Updated 2 months ago
- BLAS source of choice☆38Updated 2 weeks ago
- Coordination repository of the Machine Learning (applicant) Working Group☆18Updated 3 years ago
- Rust binding to CUDA APIs☆116Updated 2 years ago
- Generic, parallel Monte Carlo tree search library☆64Updated 2 years ago
- A Rust implementation of Gaussian Process regression.☆56Updated 8 months ago
- ☆16Updated 2 months ago
- A Rust implementation of the CMA-ES optimization algorithm.☆32Updated 2 years ago
- A Probabilistic Programming Language In Rust☆35Updated 2 years ago
- A collection of optimisers for use with candle☆29Updated last month
- Read and write tensorboard data using Rust☆16Updated 7 months ago
- ☆57Updated last year
- Low rank adaptation (LoRA) for Candle.☆124Updated 3 weeks ago
- rust-cuda working group☆65Updated 5 years ago