nelhage / transformer-rsLinks
A sketch of a Transformer in Rust for a blog post
☆32Updated 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:
- Experimentation using the xla compiler from rust☆94Updated 10 months ago
- Sample Python extension using Rust/PyO3/tch to interact with PyTorch☆37Updated last year
- Inference Llama 2 in one file of zero-dependency, zero-unsafe Rust☆38Updated last year
- Modular Rust transformer/LLM library using Candle☆36Updated last year
- ☆26Updated last year
- LLaMA from First Principles☆51Updated 2 years ago
- ☆58Updated 2 years ago
- Experimental compiler for deep learning models☆67Updated last month
- ☆130Updated last year
- Your one stop CLI for ONNX model analysis.☆47Updated 2 years ago
- LLaMa 7b with CUDA acceleration implemented in rust. Minimal GPU memory needed!☆106Updated last year
- Read and write tensorboard data using Rust☆21Updated last year
- ☆67Updated last year
- ☆23Updated last year
- Experiments for efforts to train a new and improved t5☆77Updated last year
- ☆20Updated 8 months ago
- Dataflow is a data processing library, primarily for machine learning.☆22Updated 2 years ago
- RAI: Rust ML framework with composable transformations like JAX.☆91Updated 11 months ago
- ☆32Updated 2 years ago
- implement llava using candle☆15Updated last year
- ☆39Updated 2 years ago
- A Rust Library for High-Performance Tensor Exchange with Python☆47Updated 2 weeks ago
- Generic Automatic Differentiation library for Rust (aka "autograd")☆38Updated 3 years ago
- A collection of optimisers for use with candle☆36Updated last month
- 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…☆93Updated last year
- Rust bindings to GAP (Groups, Algorithms, Programming)☆24Updated last year
- Einstein summation for Rust☆40Updated 4 years ago
- Rust wrapper for Microsoft's ONNX Runtime with CUDA support (version 1.7)☆23Updated 2 years ago
- Experimental ONNX implementation for WASI NN.☆48Updated 3 years ago
- ☆12Updated last year