herrmann / rustorch
"PyTorch in Rust"
☆16Updated last year
Alternatives and similar repositories for rustorch
Users that are interested in rustorch are comparing it to the libraries listed below
Sorting:
- Read and write tensorboard data using Rust☆21Updated last year
- Inference Llama 2 in one file of zero-dependency, zero-unsafe Rust☆38Updated last year
- Tensor library for Zig☆12Updated 6 months ago
- A small python library to run iterators in a separate process☆10Updated last year
- Because it's there.☆16Updated 7 months ago
- Make triton easier☆47Updated 11 months ago
- 👷 Build compute kernels☆38Updated this week
- ☆21Updated 5 months ago
- Sample Python extension using Rust/PyO3/tch to interact with PyTorch☆36Updated last year
- A small rust-based data loader☆24Updated 5 months ago
- ☆12Updated last year
- implement llava using candle☆14Updated 11 months ago
- ☆19Updated 7 months ago
- Rust crate for some audio utilities☆23Updated 2 months ago
- Simple (fast) transformer inference in PyTorch with torch.compile + lit-llama code☆10Updated last year
- Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.☆17Updated last month
- Experimental compiler for deep learning models☆65Updated last month
- A collection of optimisers for use with candle☆35Updated last week
- Rust Implementation of micrograd☆51Updated 10 months ago
- ☆129Updated last year
- ☆27Updated 10 months ago
- gzip Predicts Data-dependent Scaling Laws☆35Updated 11 months ago
- Code for the examples presented in the talk "Training a Llama in your backyard: fine-tuning very large models on consumer hardware" given…☆14Updated last year
- Your one stop CLI for ONNX model analysis.☆47Updated 2 years ago
- new optimizer☆19Updated 9 months ago
- ☆26Updated last year
- JAX bindings for the flash-attention3 kernels☆11Updated 9 months ago
- ☆39Updated 2 years ago
- Modular Rust transformer/LLM library using Candle☆36Updated last year
- A transformers like interface for interacting with local LLMs in Rust. This crate aims to provide the simplest interface to interact with…☆14Updated last week