2 citations · 2 across the 9 of their papers we have counts for
Showing cs.LGShow all
3 papers · 1 filter
cs.LG2025
Subquadratic Algorithms and Hardness for Attention with Any Temperature
Shreya Gupta, Boyang Huang, Barna Saha +2
Despite the popularity of the Transformer architecture, the standard algorithm for computing Attention suffers from quadratic time complexity in context length . Alman and Song…
cs.LG2024
The I/O Complexity of Attention, or How Optimal is Flash Attention?
Barna Saha, Christopher Ye
Self-attention is at the heart of the popular Transformer architecture, yet suffers from quadratic time and memory complexity. The breakthrough FlashAttention algorithm revealed I/…
cs.LG2021★ 2 cited
Predicting the Stereoselectivity of Chemical Transformations by Machine Learning
Justin Li, Dakang Zhang, Yifei Wang +3
Stereoselective reactions (both chemical and enzymatic reactions) have been essential for origin of life, evolution, human biology and medicine. Since late 1960s, there have been n…