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 2022, I've been working on NATTEN, a solution for fast multi-dimensional sparse attention (see my presentation on GPU MODE). 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 teams), through which I've been exposed to some of the most challenging problems in high performance AI.

Select publications

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.

GitHub

2024

2023

Neighborhood Attention Transformer

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

CVPR 2023.

GitHub

2022

Dilated Neighborhood Attention Transformer

Ali Hassani, and Humphrey Shi.

Preprint.

GitHub

Experience

12/2024 - 07/2025

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

Spring 2025

TA, CS 4476: Intro to Computer Vision

Fall 2023

Instructor, CS 322: Intro to Software Engineering

Spring 2021, Winter 2022, Winter 2023

Education

2024 - 2026 (expected)

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