Ali Hassani

Computer Science PhD student at Georgia Tech

I am a computer science PhD student at Georgia Tech advised by Prof. Humphrey Shi. My primary research area is High Performance AI Architecture, which involves making fast and efficient AI, from neural network architecture down to computer architecture. Since 2021, I've been working on NATTEN, a solution for fast multi-dimensional sparse attention. In our most recent paper, Generalized Neighborhood Attention, we discuss outstanding challenges in sparse attention infrastructure, and our proposed solutions, which involve both an analytical study, and a fast implementation achieving the maximum speedup theoretically possible. I've been fortunate to have worked at HippoML and NVIDIA (both in production and research), through which I've been exposed to some of the most challenging problems in high performance AI.

GitHub / Google Scholar / Twitter / LinkedIn

Select publications & preprints

2025

Generalized Neighborhood Attention: Multi-dimensional Sparse Attention at the Speed of Light

Ali Hassani, Fengzhe Zhou, Aditya Kane, Jiannan Huang, Chieh-Yun Chen, Min Shi, Steven Walton, Markus Hoehnerbach, Vijay Thakkar, Michael Isaev, Qinsheng Zhang, Bing Xu, Haicheng Wu, Wen-mei Hwu, Ming-Yu Liu, and Humphrey Shi.

Preprint.

2024

2023

Neighborhood Attention Transformer

Ali Hassani, Steven Walton, Jiachen Li, Shen Li, and Humphrey Shi.

CVPR 2023.

2022

Dilated Neighborhood Attention Transformer

Ali Hassani, and Humphrey Shi.

Preprint.

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

Since 12/2024

05/2024 - 08/2024

NVIDIA, CUTLASS

Software Performance Engineering Intern

06/2023 - 12/2023

HippoML

Software Engineering Intern

06/2022 - 09/2022

Picsart AI Research

Research Intern

Since 2024

SHI Labs at Georgia Tech

Graduate Researcher

2021-2023

SHI Labs at University of Oregon

Graduate Researcher

2019 to 2021

University of Kerman, Mahani Mathematical Research Center

Undergraduate Researcher

Teaching

S25

TA, CS 4476: Intro to Computer Vision

F23

Instructor, CS 322: Intro to Software Engineering

S21, W22, W23

Education

Since 2024

PhD in Computer Science

Advised by Prof. Humphrey Shi

2021-2023

M.S. in Computer Science

Advised by Prof. Humphrey Shi

2016-2020

B.S. in Computer Science

Research advisor: Prof. Abbas Salemi

University of Kerman