Bokun Wang

Postdoctoral Researcher at UT Austin

I am a Postdoctoral Fellow in the Department of Electrical and Computer Engineering (ECE) at The University of Texas at Austin, working with Prof. Diana Marculescu and the EnyAC group. Before that, I received my Ph.D. degree in Computer Science at Texas A&M University, advised by Prof. Tianbao Yang.

My research interest lies in the Hardware-efficient Generative AI.

Research

This project focuses on improving hardware efficiency (latency, memory, energy, etc.) of video and language GenAI inference.

LIPAR ELANA

Developing efficient algorithms for (multi-modal) self-supervised representation learning (SSL) with convergence and generalization guarantees.

SCENT GloFND DPM SOX LibAUC

This project focuses on efficient and provable algorithms for high-resolution, imbalanced, and multi-instance medical imaging data classification.

MIDAM SOPA

This project focuses on developing communication-efficient and provably convergent algorithms for distributed and federated learning.

FLFP8 MOML IntSGD

Dennis Menn, Yuedong Yang, Bokun Wang, Xiwen Wei, Mustafa Munir, Feng Liang, Radu Marculescu, Chenfeng Xu, and Diana Marculescu

ArXiv Preprint, 2026.

[PDF] [Code] [Webpage]

Hung-Yueh Chiang, Bokun Wang, and Diana Marculescu

Technical Report, 2025.

[PDF] [Webpage]

Vicente Balmaseda, Bokun Wang, Ching-Long Lin, and Tianbao Yang

In Proceedings of the Forty-second International Conference on Machine Learning (ICML), 2025.

[PDF] [Code] [Poster]

Guest Lecturer, CSCE 689: Optimization for Machine Learning (Fall 2023), Texas A&M University.
Teaching Assistant, ECS 32B: Introduction to Data Structures (Winter 2020), University of California Davis.
Teaching Assistant, ECS 154A: Computer Architecture (Fall 2019), University of California Davis.
Teaching Assistant, ECS 170: Introduction to Artificial Intelligence (Spring 2019), University of California Davis.
Teaching Assistant, ECS 271: Machine Learning and Data Discovery (Winter 2019), University of California Davis.