Alternatives and similar repositories for tch-ext

Users that are interested in tch-ext are comparing it to the libraries listed below

• LaurentMazare / xla-rs
  Experimentation using the xla compiler from rust
  ☆99Updated last year
• LaurentMazare / ug
  Experimental compiler for deep learning models
  ☆71Updated 2 months ago
• LaurentMazare / tboard-rs
  Read and write tensorboard data using Rust
  ☆23Updated last year
• mokeyish / candle-ext
  An extension library to Candle that provides PyTorch functions not currently available in Candle
  ☆40Updated last year
• sarah-quinones / gemm
  ☆96Updated 3 weeks ago
• EricLBuehler / float8
  8-bit floating point types for Rust
  ☆61Updated this week
• Narsil / bindgen_cuda
  ☆26Updated 7 months ago
• cksac / rai
  RAI: Rust ML framework with composable transformations like JAX.
  ☆93Updated last year
• Tudyx / ai-dataloader
  A rust port of pytorch dataloader
  ☆29Updated 11 months ago
• LaurentMazare / glim
  ☆19Updated last year
• SunDoge / dlpark
  A Rust Library for High-Performance Tensor Exchange with Python
  ☆48Updated last week
• KGrewal1 / candle-optimisers
  A collection of optimisers for use with candle
  ☆44Updated this week
• EricLBuehler / candle-lora
  Low rank adaptation (LoRA) for Candle.
  ☆168Updated 7 months ago
• tomsanbear / candle-einops
  ☆36Updated last year
• Narsil / ggblas
  ☆28Updated 2 years ago
• KarelPeeters / Kyanite
  A neural network inference library, written in Rust.
  ☆70Updated last year
• RustyBamboo / rust-grad
  Automatic differentiation in Rust with WGPU support
  ☆24Updated 4 years ago
• elftausend / custos
  A minimal OpenCL, CUDA, Vulkan and host CPU array manipulation engine / framework.
  ☆76Updated 3 months ago
• chelsea0x3b / llama-dfdx
  LLaMa 7b with CUDA acceleration implemented in rust. Minimal GPU memory needed!
  ☆110Updated 2 years ago
• eugenehp / gpu-fft
  GPU based FFT written in Rust and CubeCL
  ☆24Updated 5 months ago
• Narsil / smelte-rs
  ☆58Updated 2 years ago
• huggingface / hf-hub
  Rust client for the huggingface hub aiming for minimal subset of features over `huggingface-hub` python package
  ☆243Updated 2 weeks ago
• drbh / simd-alphatensor-rs
  🧮 alphatensor matrix breakthrough algorithms + simd + rust.
  ☆62Updated 3 years ago
• jafioti / dataflow
  Dataflow is a data processing library, primarily for machine learning.
  ☆24Updated 2 years ago
• gmmyung / eerie
  Rustic bindings to the IREE Compiler/Runtime
  ☆24Updated 3 months ago
• haixuanTao / onnxruntime-rs
  Rust wrapper for Microsoft's ONNX Runtime with CUDA support (version 1.7)
  ☆24Updated 3 years ago
• ssoudan / tch-m1
  Example of tch-rs on M1
  ☆55Updated last year
• ritchie46 / lsh-rs
  Locality Sensitive Hashing in Rust with Python bindings
  ☆119Updated 2 years ago
• Mathemmagician / rustygrad
  Tiny Autograd engine written in Rust
  ☆61Updated last year
• blas-lapack-rs / openblas-src
  Source of BLAS and LAPACK via OpenBLAS
  ☆89Updated 2 months ago