publications and pre-prints

publications by categories in reversed chronological order. *,^ denotes equal author contribution.

2025

  1. NLP
    kernelbench.png
    KernelBench: Can LLMs Write Efficient GPU Kernels?
    Anne Ouyang*, Simon Guo*, Simran Arora, Alex Zhang , and 3 more authors
    DL4C (Best Paper Award), SSI-FM Workshop @ ICLR 2025
    arXiv Code

2024

  1. NLP
    teaser_mm.png
    SWE-bench Multimodal: Do AI Systems Generalize to Visual Software Domains?
    John Yang*, Carlos E. Jimenez*, Alex Zhang, Kilian Lieret , and 9 more authors
    ICLR 2025
    arXiv Website
  2. NLP
    lgwm.png
    Language-guided World Models: A Model-Based Approach to AI Control.
    Alex Zhang*, Khanh Nguyen*, Jens Tuyls, Albert Lin , and 1 more author
    SpLU-RoboNLP Workshop @ ACL 2024 (Oral)
    arXiv Code Website

2023

  1. Dataset
    asap.png
    Building Scalable Video Understanding Benchmarks through Sports
    Aniket Agarwal*, Alex Zhang*, Karthik Narasimhan, Igor Gilitschenski , and 2 more authors
    DMLR Workshop @ ICLR 2024
    arXiv Code Website
  2. Survey
    Transaction Fee Mining and Mechanism Design
    Michael Tang*, and Alex Zhang*
    arXiv

2019

  1. RL
    tremor.png
    Developing a Device- Independent Software for Adaptive Tremor Suppression Orthoses
    Alex Zhang
    HTML