Dr. Sasha Rush
Cornell Tech / Hugging Face
Verified Proof of Work Artifacts
2 items catalogedEach artifact below represents an authenticated research publication, production code repository, or technical architectural framework directly authored or co-created by Dr. Sasha Rush. Every entry undergoes editorial source verification.
Flash-Decoding: Accelerating Long-Context LLM Inference via Partitioned Attention
Co-developed Flash-Decoding, an inference parallelization algorithm that partitions key-value caches across thread blocks during token generation, delivering up to 8x speedup on 64k+ context sequences.
Requires reduction step across partitions that introduces minor overhead for very short sequence lengths.
The Annotated Transformer: A Line-by-Line PyTorch Implementation
Authored the universally referenced educational codebase providing an exact, executable line-by-line PyTorch implementation of 'Attention Is All You Need', bridging theoretical tensor math with production code.
Presents vanilla quadratic multi-head attention without later FlashAttention or grouped-query optimizations.