Alternatives and similar repositories for tch-ext:

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

• LaurentMazare / tboard-rs
  Read and write tensorboard data using Rust
  ☆19Updated last year
• KGrewal1 / candle-optimisers
  A collection of optimisers for use with candle
  ☆31Updated 2 months ago
• LaurentMazare / xla-rs
  Experimentation using the xla compiler from rust
  ☆91Updated 5 months ago
• Narsil / bindgen_cuda
  ☆21Updated 7 months ago
• LaurentMazare / glim
  ☆18Updated 4 months ago
• Narsil / ggblas
  ☆25Updated last year
• LaurentMazare / ug
  Experimental compiler for deep learning models
  ☆26Updated last month
• cksac / rai
  RAI: Rust ML framework with composable transformations like JAX.
  ☆84Updated 6 months ago
• LaurentMazare / hojo
  A small python library to run iterators in a separate process
  ☆10Updated 10 months ago
• huggingface / candle-paged-attention
  ☆12Updated last year
• ssoudan / tch-m1
  Example of tch-rs on M1
  ☆51Updated 10 months ago
• mokeyish / candle-ext
  An extension library to Candle that provides PyTorch functions not currently available in Candle
  ☆38Updated 11 months ago
• tomsanbear / candle-einops
  ☆27Updated 2 months ago
• oxideai / diffusers-burn
  A diffusers API in Burn (Rust)
  ☆19Updated 7 months ago
• EricLBuehler / float8
  8-bit floating point types for Rust
  ☆44Updated 2 weeks ago
• SunDoge / dlpark
  A Rust Library for High-Performance Tensor Exchange with Python
  ☆39Updated 2 months ago
• EricLBuehler / candle-lora
  Low rank adaptation (LoRA) for Candle.
  ☆141Updated 5 months ago
• sarah-quinones / gemm
  ☆81Updated last month
• SamKG / iree-rs
  Rustic bindings for IREE
  ☆18Updated last year
• coreylowman / llama-dfdx
  LLaMa 7b with CUDA acceleration implemented in rust. Minimal GPU memory needed!
  ☆102Updated last year
• RustyBamboo / rust-grad
  Automatic differentiation in Rust with WGPU support
  ☆23Updated 3 years ago
• oracleofnj / einsum
  Einstein summation for Rust
  ☆35Updated 3 years ago
• chenwanqq / candle-llava
  implement llava using candle
  ☆14Updated 8 months ago
• oddity-ai / async-cuda
  Asynchronous CUDA for Rust.
  ☆29Updated 3 months ago
• EricLBuehler / candle_graphs
  Graph model execution API for Candle
  ☆12Updated 2 months ago
• nelhage / transformer-rs
  A sketch of a Transformer in Rust for a blog post
  ☆29Updated 2 years ago
• elftausend / custos
  A minimal OpenCL, CUDA, Vulkan and host CPU array manipulation engine / framework.
  ☆72Updated 2 months ago
• huggingface / hf-hub
  Rust client for the huggingface hub aiming for minimal subset of features over `huggingface-hub` python package
  ☆178Updated last month
• zk4x / zyx
  Tensor library for machine learning
  ☆24Updated this week
• BDUG / Lighter
  A Keras like abstraction layer on top of the Rust ML framework candle
  ☆23Updated 8 months ago