I am a computer science PhD student at Georgia Tech, and am advised by Prof. Humphrey Shi. I received my master's degree from University of Oregon under Prof. Shi, and my undergraduate from University of Kerman, where I did research under Prof. Abbas Salemi. I'm interested in a wide range of topics concerning Deep Learning, Computer Vision, and High Performance Computing. My research started out with classical unsupervised ML methods during my undergraduate, and deep learning and computer vision in PhD, and shifted into machine learning systems, system co-design, and high performance computing. I've been fortunate to have interned at HippoML (under Bing Xu) and NVIDIA (mentored by Haicheng Wu). My goal is to contribute to help scale up current and future AI and general-purpose computing systems. My longest running research project is Neighborhood Attention, a multi-axis sliding window attention pattern that allows fine-grained control over attention span, drawing a spectrum of possible attention patterns between linear projection and self attention. Most of my research has involved designing deep learning architectures based on Neighborhood Attention, scaling them up, and most importantly, accelerating it for general-purpose accelerators, namely NVIDIA GPUs. Check out our open-source project NATTEN for more details. I've also contributed to open source projects such as CUTLASS.
GitHub / Google Scholar / Twitter / LinkedIn
Select publications & preprints
2024
Faster Neighborhood Attention: Reducing the O(n^2) Cost of Self Attention at the Threadblock Level
Ali Hassani, Wen-Mei Hwu, and Humphrey Shi.
To appear in NeurIPS 2024.
2023
Neighborhood Attention Transformer
Ali Hassani, Steven Walton, Jiachen Li, Shen Li, and Humphrey Shi.
CVPR 2023.
2022
StyleNAT: Giving Each Head a New Perspective
Steven Walton, Ali Hassani, Xingqian Xu, Zhangyang Wang, and Humphrey Shi.
Preprint.
2021
Escaping the Big Data Paradigm with Compact Transformers
Ali Hassani*, Steven Walton*, Nikhil Shah, Abulikemu Abuduweili, Jiachen Li, and Humphrey Shi.
Preprint.
Experience
05/2024 - 08/2024
Performance Software Engineering Intern
06/2023 - 12/2023
Software Engineering Intern
06/2022 - 09/2022
Picsart AI Research
Research Intern
Since 03/2021
Graduate Researcher
2019 to 2021
Mahani Mathematical Research Center
Undergraduate Researcher
Teaching
Education
2016-2020
B.S. in Computer Science
University of Kerman