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
  ☆21Updated last year
• LaurentMazare / xla-rs
  Experimentation using the xla compiler from rust
  ☆93Updated 9 months ago
• LaurentMazare / glim
  ☆19Updated 7 months ago
• KGrewal1 / candle-optimisers
  A collection of optimisers for use with candle
  ☆36Updated 2 weeks ago
• Narsil / ggblas
  ☆26Updated last year
• LaurentMazare / ug
  Experimental compiler for deep learning models
  ☆67Updated last week
• EricLBuehler / float8
  8-bit floating point types for Rust
  ☆45Updated 2 months ago
• mokeyish / candle-ext
  An extension library to Candle that provides PyTorch functions not currently available in Candle
  ☆39Updated last year
• Narsil / bindgen_cuda
  ☆23Updated last month
• jafioti / dataflow
  Dataflow is a data processing library, primarily for machine learning.
  ☆21Updated last year
• huggingface / candle-paged-attention
  ☆12Updated last year
• sarah-quinones / gemm
  ☆86Updated 4 months ago
• cksac / rai
  RAI: Rust ML framework with composable transformations like JAX.
  ☆89Updated 10 months ago
• LaurentMazare / hojo
  A small python library to run iterators in a separate process
  ☆10Updated last year
• SunDoge / dlpark
  A Rust Library for High-Performance Tensor Exchange with Python
  ☆45Updated this week
• tomsanbear / candle-einops
  ☆30Updated 6 months ago
• EricLBuehler / candle-lora
  Low rank adaptation (LoRA) for Candle.
  ☆147Updated last month
• VasanthakumarV / einops
  Simplistic API for deep learning tensor operations
  ☆30Updated last month
• chenwanqq / candle-llava
  implement llava using candle
  ☆15Updated 11 months ago
• oracleofnj / einsum
  Einstein summation for Rust
  ☆38Updated 4 years ago
• KarelPeeters / Kyanite
  A neural network inference library, written in Rust.
  ☆62Updated 10 months ago
• coreylowman / llama-dfdx
  LLaMa 7b with CUDA acceleration implemented in rust. Minimal GPU memory needed!
  ☆104Updated last year
• ssoudan / tch-m1
  Example of tch-rs on M1
  ☆53Updated last year
• jeroenvlek / gpt-from-scratch-rs
  Andrej Karpathy's Let's build GPT: from scratch video & notebook implemented in Rust + candle
  ☆71Updated last year
• Tudyx / ai-dataloader
  A rust port of pytorch dataloader
  ☆26Updated 5 months ago
• mrlazy1708 / tqdm
  Python tqdm in Rust.
  ☆45Updated last year
• eugenehp / gpu-fft
  GPU based FFT written in Rust and CubeCL
  ☆22Updated 2 months ago
• LaurentMazare / mamba.rs
  ☆129Updated last year
• elftausend / custos
  A minimal OpenCL, CUDA, Vulkan and host CPU array manipulation engine / framework.
  ☆73Updated last week
• KerfuffleV2 / rusty-ggml
  GGML bindings that aim to be idiomatic Rust rather than directly corresponding to the C/C++ interface
  ☆19Updated last year