srush

Sasha Rush's repositories

GPU-Puzzles

Solve puzzles. Learn CUDA.

Language:Jupyter NotebookMIT5018 27 25

Tensor-Puzzles

Solve puzzles. Improve your pytorch.

Language:Jupyter NotebookMIT2447 10 17

MiniChain

A tiny library for coding with large language models.

Language:PythonMIT1170 15 10

llama2.rs

A fast llama2 decoder in pure Rust.

Language:RustMIT978 10 20

LLM-Training-Puzzles

What would you do with 1000 H100s...

Language:Jupyter NotebookMIT731 12 3

Triton-Puzzles

Puzzles for learning Triton

Language:Jupyter NotebookApache-2.0626 3 6

annotated-mamba

Annotated version of the Mamba paper

Language:Jupyter NotebookMIT397 23 2

Autodiff-Puzzles

Language:Jupyter NotebookMIT292 6 4

Transformer-Puzzles

Puzzles for exploring transformers

Language:Jupyter NotebookMIT254 7 1

raspy

An interactive exploration of Transformer programming.

Language:Jupyter NotebookMIT226 6 2

do-we-need-attention

Language:TeXMIT160 8 1

GPTWorld

A puzzle to learn about prompting

Language:Jupyter NotebookMIT96 4 1

triton-autodiff

Experiment of using Tangent to autodiff triton

Language:PythonMIT66 40

torch-queue

Language:PythonMIT62 30

LLM-Talk

43 3 1

mamba-primer

29 30

drop7

Language:Jupyter Notebook18 20

g9py

Language:HTML18 20

mamba-scans

Blog post

16 4 1

tangent

Source-to-Source Debuggable Derivatives in Pure Python

Language:PythonApache-2.014 10

Tensor-Puzzles-Penzai

Language:HTML1000

minitorch-rust

MIT7 70

PowerEdit

A super-minimal Python-based video editor ⚡

Language:PythonMIT4 20

SimpliftDemo

Language:JavaScript2 10

triton

MIT2 30

osgai

Language:JavaScriptMIT1 20

triton-edit

Development repository for the Triton language and compiler

Language:C++MIT1 10

gradio-tools

Language:PythonMIT010

staging

Language:HTMLMIT010

triton-scans

020