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.
awesome-o1
A bibliography and survey of the papers surrounding o1
GPU-Puzzles
Solve puzzles. Learn CUDA.
llama2.rs
A fast llama2 decoder in pure Rust.
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/Triton-Puzzles
Puzzles for learning Triton
srush/MiniChain
A tiny library for coding with large language models.
srush/awesome-o1
A bibliography and survey of the papers surrounding o1
srush/llama2.rs
A fast llama2 decoder in pure Rust.
srush/LLM-Training-Puzzles
What would you do with 1000 H100s...
srush/annotated-mamba
Annotated version of the Mamba paper
srush/Autodiff-Puzzles
srush/raspy
An interactive exploration of Transformer programming.
srush/triton-autodiff
Experiment of using Tangent to autodiff triton
srush/torch-queue
srush/LLM-Talk
srush/mamba-primer
srush/prof8
Experimental paper writing linter.
srush/Tensor-Puzzles-Penzai
srush/anynp
Proof-of-concept of global switching between numpy/jax/pytorch in a library.
srush/drop7
srush/g9py
srush/mamba-scans
Blog post
srush/tangent
Source-to-Source Debuggable Derivatives in Pure Python
srush/DiffRast
srush/PowerEdit
A super-minimal Python-based video editor ⚡
srush/colm-search
srush/SimpliftDemo
srush/wgpu-shaders
srush/osgai
srush/triton-edit
Development repository for the Triton language and compiler
srush/srush.github.io
srush/triton-scans