← Search

Genghan Zhang

2 accepted papers

2025

Adaptive Self-improvement LLM Agentic System for ML Library Development

ICML 2025poster

ML libraries, often written in architecture-specific programming languages (ASPLs) that target domain-specific architectures, are key to efficient ML systems. However, writing these high-performance ML libraries is challenging because it requires expert knowledge of both ML algorithms and the ASPL.…

2025

Learning to (Learn at Test Time): RNNs with Expressive Hidden States

ICML 2025spotlight

Self-attention performs well in long context but has quadratic complexity. Existing RNN layers have linear complexity, but their performance in long context is limited by the expressive power of their hidden states. We present a practical framework for instantiating sequence modeling layers with lin…