Ali Hassani

Computer science PhD student at Georgia Tech

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

2023

2022

2022

2021

Experience

05/2024 - 08/2024

NVIDIA

Performance Software Engineering Intern

06/2023 - 12/2023

HippoML

Software Engineering Intern

06/2022 - 09/2022

Picsart AI Research

Research Intern

Since 03/2021

SHI Labs

Graduate Researcher

2019 to 2021

Mahani Mathematical Research Center

Undergraduate Researcher

Teaching

F23

Instructor, CS 322: Intro to Software Engineering

S21, W22, W23

Education

Ongoing

PhD in Computer Science

2021-2023

M.S. in Computer Science

2016-2020

B.S. in Computer Science

University of Kerman