herrmann / rustorchLinks
"PyTorch in Rust"
☆17Updated last year
Alternatives and similar repositories for rustorch
Users that are interested in rustorch are comparing it to the libraries listed below
Sorting:
- Rust crate for some audio utilities☆25Updated 9 months ago
- Read and write tensorboard data using Rust☆24Updated last year
- ☆39Updated 3 years ago
- Rust Implementation of micrograd☆53Updated last year
- A collection of optimisers for use with candle☆44Updated this week
- Training hybrid models for dummies.☆29Updated last month
- implement llava using candle☆15Updated last year
- ☆135Updated last year
- ☆24Updated 11 months ago
- ☆16Updated last year
- See https://github.com/cuda-mode/triton-index/ instead!☆11Updated last year
- Utilities for Training Very Large Models☆58Updated last year
- LayerNorm(SmallInit(Embedding)) in a Transformer to improve convergence☆61Updated 3 years ago
- PyTorch Implementation of the paper "MM1: Methods, Analysis & Insights from Multimodal LLM Pre-training"☆24Updated last week
- Efficiently computing & storing token n-grams from large corpora☆26Updated last year
- Implementation of Hyena Hierarchy in JAX☆10Updated 2 years ago
- Exploring an idea where one forgets about efficiency and carries out attention across each edge of the nodes (tokens)☆55Updated 8 months ago
- Sample Python extension using Rust/PyO3/tch to interact with PyTorch☆39Updated last year
- Rust bindings for CTranslate2☆14Updated 2 years ago
- Make triton easier☆49Updated last year
- JAX/Flax implementation of the Hyena Hierarchy☆34Updated 2 years ago
- JAX bindings for the flash-attention3 kernels☆16Updated 2 months ago
- Collection of autoregressive model implementation☆85Updated 7 months ago
- Modular Rust transformer/LLM library using Candle☆37Updated last year
- Code for the examples presented in the talk "Training a Llama in your backyard: fine-tuning very large models on consumer hardware" given…☆14Updated 2 years ago
- CUDA and Triton implementations of Flash Attention with SoftmaxN.☆73Updated last year
- JAX Scalify: end-to-end scaled arithmetics☆17Updated last year
- ☆12Updated last year
- Engineering the state of RNN language models (Mamba, RWKV, etc.)☆32Updated last year
- RWKV model implementation☆38Updated 2 years ago