Pinned Repositories
annotated-transformer
An annotated implementation of the Transformer paper.
pytorch-struct
Fast, general, and tested differentiable structured prediction in PyTorch
LSTMVis
Visualization Toolbox for Long Short Term Memory networks (LSTMs)
transformers
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
GPU-Puzzles
Solve puzzles. Learn CUDA.
llama2.rs
A fast llama2 decoder in pure Rust.
LLM-Training-Puzzles
What would you do with 1000 H100s...
MiniChain
A tiny library for coding with large language models.
Tensor-Puzzles
Solve puzzles. Improve your pytorch.
Triton-Puzzles
Puzzles for learning Triton
srush's Repositories
srush/GPU-Puzzles
Solve puzzles. Learn CUDA.
srush/Tensor-Puzzles
Solve puzzles. Improve your pytorch.
srush/MiniChain
A tiny library for coding with large language models.
srush/llama2.rs
A fast llama2 decoder in pure Rust.
srush/LLM-Training-Puzzles
What would you do with 1000 H100s...
srush/Triton-Puzzles
Puzzles for learning Triton
srush/annotated-mamba
Annotated version of the Mamba paper
srush/Autodiff-Puzzles
srush/Transformer-Puzzles
Puzzles for exploring transformers
srush/raspy
An interactive exploration of Transformer programming.
srush/do-we-need-attention
srush/GPTWorld
A puzzle to learn about prompting
srush/triton-autodiff
Experiment of using Tangent to autodiff triton
srush/torch-queue
srush/LLM-Talk
srush/mamba-primer
srush/g9py
srush/mamba-scans
Blog post
srush/tangent
Source-to-Source Debuggable Derivatives in Pure Python
srush/drop7
srush/minitorch-rust
srush/PowerEdit
A super-minimal Python-based video editor ⚡
srush/SimpliftDemo
srush/triton
srush/osgai
srush/triton-edit
Development repository for the Triton language and compiler
srush/gradio-tools
srush/openai-plugins-tool
srush/staging
srush/triton-scans