Publications

You can also find my articles on my Google Scholar profile.

Conference Papers

The Surprising Soupability of Documents in State Space Models

Published in arXiv, 2025

Investigating document mixing in state space models

Recommended citation: Y. Jafari, Z. Wang, L. Bergen, T. Berg-Kirkpatrick. "The Surprising Soupability of Documents in State Space Models." arXiv preprint arXiv:2505.24033.
Download Paper

Quiet Feature Learning in Algorithmic Tasks

Published in AAAI 2026, 2025

Oral presentation at AAAI 2026 Special Track on AI Alignment

Recommended citation: P. Naidu, Z. Wang, L. Bergen, R. Paturi. "Quiet Feature Learning in Algorithmic Tasks." AAAI 2026 Special Track on AI Alignment.
Download Paper

Omniwise: Predicting GPU Kernels Performance with LLMs

Published in arXiv, 2025

Using large language models to predict GPU kernel performance

Recommended citation: Z. Wang, C. Ramos, M. A. Awad, K. Lowery. "Omniwise: Predicting GPU Kernels Performance with LLMs." arXiv preprint arXiv:2506.20886.
Download Paper

Preliminary Results of the MLPerf BERT Inference Benchmark on AMD Instinct GPUs

Published in Practice and Experience in Advanced Research Computing 2024, 2024

MLPerf BERT Inference Benchmark results on AMD Instinct GPUs

Recommended citation: Z. Wang, K. Vu, M. Hodak, A. Mehrotra, F. Gutierrez, K. Smith, G. Seo, et al. "Preliminary Results of the MLPerf BERT Inference Benchmark on AMD Instinct GPUs." Practice and Experience in Advanced Research Computing 2024: Human Powered Computing.