nelhage / transformer-rsLinks
A sketch of a Transformer in Rust for a blog post
☆34Updated 3 years ago
Alternatives and similar repositories for transformer-rs
Users that are interested in transformer-rs are comparing it to the libraries listed below
Sorting:
- Modular Rust transformer/LLM library using Candle☆37Updated last year
- ☆58Updated 2 years ago
- ☆135Updated last year
- LLaMA from First Principles☆51Updated 2 years ago
- Your one stop CLI for ONNX model analysis.☆47Updated 3 years ago
- Inference Llama 2 in one file of zero-dependency, zero-unsafe Rust☆39Updated 2 years ago
- Sample Python extension using Rust/PyO3/tch to interact with PyTorch☆39Updated last year
- Experimental compiler for deep learning models☆71Updated 2 months ago
- Experimentation using the xla compiler from rust☆99Updated last year
- ☆157Updated 2 years ago
- Inference Llama 2 in one file of pure Rust 🦀☆235Updated 2 years ago
- ☆32Updated 2 years ago
- Dataflow is a data processing library, primarily for machine learning.☆24Updated 2 years ago
- A relatively basic implementation of RWKV in Rust written by someone with very little math and ML knowledge. Supports 32, 8 and 4 bit eva…☆94Updated 2 years ago
- Locality Sensitive Hashing in Rust with Python bindings☆119Updated 2 years ago
- ☆19Updated last year
- Tiny Autograd engine written in Rust☆61Updated last year
- LayerNorm(SmallInit(Embedding)) in a Transformer to improve convergence☆61Updated 3 years ago
- Using Large Language Models for Repo-wide Type Prediction☆112Updated 2 years ago
- Tensor library with autograd using only Rust's standard library☆70Updated last year
- implement llava using candle☆15Updated last year
- A high-performance constrained decoding engine based on context free grammar in Rust☆56Updated 6 months ago
- Neural syntax annotator, supporting sequence labeling, lemmatization, and dependency parsing.☆78Updated 2 years ago
- Rust wrapper for Microsoft's ONNX Runtime with CUDA support (version 1.7)☆24Updated 3 years ago
- ☆23Updated 2 years ago
- RAI: Rust ML framework with composable transformations like JAX.☆93Updated last year
- Experiments for efforts to train a new and improved t5☆76Updated last year
- Doing advent of code with CUDA and rust.☆205Updated last year
- HomebrewNLP in JAX flavour for maintable TPU-Training☆51Updated last year
- Experimental ONNX implementation for WASI NN.☆48Updated 4 years ago