Alternatives and similar repositories for xla-rs:

Users that are interested in xla-rs are comparing it to the libraries listed below

• cksac / rai
  RAI: Rust ML framework with composable transformations like JAX.
  ☆85Updated 8 months ago
• LaurentMazare / tch-ext
  Sample Python extension using Rust/PyO3/tch to interact with PyTorch
  ☆34Updated last year
• sarah-quinones / gemm
  ☆84Updated 2 months ago
• EnzymeAD / rust
  A rust fork to work towards Enzyme integration
  ☆83Updated this week
• EnzymeAD / oxide-enzyme
  Enzyme integration into Rust. Experimental, do not use.
  ☆102Updated 2 years ago
• SunDoge / dlpark
  A Rust Library for High-Performance Tensor Exchange with Python
  ☆40Updated 4 months ago
• RustyBamboo / rust-grad
  Automatic differentiation in Rust with WGPU support
  ☆23Updated 3 years ago
• elrnv / autodiff
  Auto differentiation library
  ☆68Updated last year
• LaurentMazare / ug
  Experimental compiler for deep learning models
  ☆28Updated last month
• LaurentMazare / tboard-rs
  Read and write tensorboard data using Rust
  ☆19Updated last year
• SamKG / iree-rs
  Rustic bindings for IREE
  ☆18Updated 2 years ago
• Narsil / bindgen_cuda
  ☆21Updated 8 months ago
• mokeyish / candle-ext
  An extension library to Candle that provides PyTorch functions not currently available in Candle
  ☆38Updated last year
• oracleofnj / einsum
  Einstein summation for Rust
  ☆36Updated 3 years ago
• KarelPeeters / Kyanite
  A neural network inference library, written in Rust.
  ☆60Updated 8 months ago
• Narsil / ggblas
  ☆25Updated last year
• pengowen123 / cmaes
  A Rust implementation of the CMA-ES optimization algorithm.
  ☆35Updated 3 months ago
• ssoudan / tch-m1
  Example of tch-rs on M1
  ☆53Updated last year
• blas-lapack-rs / blas-src
  BLAS source of choice
  ☆41Updated last month
• zxqfl / mcts
  Generic, parallel Monte Carlo tree search library
  ☆72Updated 2 years ago
• LaurentMazare / glim
  ☆19Updated 5 months ago
• mlir-rs / mlir-sys
  Rust bindings to the MLIR C API.
  ☆65Updated last month
• coreylowman / llama-dfdx
  LLaMa 7b with CUDA acceleration implemented in rust. Minimal GPU memory needed!
  ☆104Updated last year
• tomsanbear / candle-einops
  ☆28Updated 4 months ago
• termoshtt / link_cuda_kernel
  HowTo: Compile CUDA with nvcc, and link to Rust
  ☆48Updated 6 years ago
• Narsil / smelte-rs
  ☆57Updated last year
• jturner314 / ndarray-npy
  .npy and .npz file format support for ndarray
  ☆59Updated 6 months ago
• nestordemeure / friedrich
  A Rust implementation of Gaussian Process regression.
  ☆57Updated last year
• bluss / matrixmultiply
  General matrix multiplication of f32 and f64 matrices in Rust. Supports matrices with general strides.
  ☆229Updated 7 months ago
• coreylowman / synthesis
  A rust implementation of AlphaZero algorithm
  ☆47Updated 2 years ago