Yida Wang

Principal Researcher, Amazon

Joined

July 2019

Names

Yida Wang

Emails

****@amazon.com (Confirmed)

Personal Links

Career & Education History

Principal Researcher

Amazon (amazon.com)

2017 – Present

Advisors, Relations & Conflicts

No relations added

Expertise

No areas of expertise listed

Publications

AccelOpt: A Self-Improving LLM Agentic System for AI Accelerator Kernel Optimization
Genghan Zhang, Shaowei Zhu, Anjiang Wei, Zhenyu Song, Allen Nie, Zhen Jia, Nandita Vijaykumar, Yida Wang, Kunle Olukotun
- LLA 2026 Poster
- Readers: Everyone
AccelOpt: A Self-Improving LLM Agentic System for AI Accelerator Kernel Optimization
Genghan Zhang, Shaowei Zhu, Anjiang Wei, Zhenyu Song, Allen Nie, Zhen Jia, Nandita Vijaykumar, Yida Wang, Kunle Olukotun
- MALGAI
- Readers: Everyone
DIFFUSIONPIPE: TRAINING LARGE DIFFUSION MODELS WITH EFFICIENT PIPELINES
Ye Tian, Zhen Jia, Ziyue Luo, Yida Wang, Chuan Wu
- OpenReview Archive Direct Upload
- Readers: Everyone
AccelOpt: A Self-Improving LLM Agentic System for AI Accelerator Kernel Optimization
Genghan Zhang, Shaowei Zhu, Anjiang Wei, Zhenyu Song, Allen Nie, Zhen Jia, Nandita Vijaykumar, Yida Wang, Kunle Olukotun
- Architecture 2.0 2026
- Readers: Everyone
EmbedMol: An Open Billion-scale Molecular Embedding Dataset for Molecular Discovery
Zhiyao Tang, Yida Wang, Jing Li
- ICLR 2026 Conference Withdrawn Submission
- Readers: Everyone
PLoRA: Efficient LoRA Hyperparameter Tuning for Large Language Models
Minghao Yan, Zhuang Wang, Zhen Jia, Shivaram Venkataraman, Yida Wang
- Submitted to ICLR 2026
- Readers: Everyone
TritonRL: Training LLMs to Think and Code Triton Without Cheating
Jiin Woo, Shaowei Zhu, Allen Nie, Zhen Jia, Yida Wang, Youngsuk Park
- Submitted to ICLR 2026
- Readers: Everyone
KVFlow: Efficient Prefix Caching for Accelerating LLM-Based Multi-Agent Workflows
Zaifeng Pan, AJJKUMAR PATEL, Yipeng Shen, Zhengding Hu, Yue Guan, Wan-Lu Li, Lianhui Qin, Yida Wang, Yufei Ding
- NeurIPS 2025 poster
- Readers: Everyone
Block-Diagonal LoRA for Eliminating Communication Overhead in Tensor Parallel LoRA Serving
Xinyu Wang, Jonas M. Kübler, Kailash Budhathoki, Yida Wang, Matthäus Kleindessner
- NeurIPS 2025 poster
- Readers: Everyone
ScaleFusion: Scalable Inference of Spatial-Temporal Diffusion Transformers for High-Resolution Long Video Generation
Jiacheng Yang, Jun Wu, Zhen Zhang, Xinwei Fu, Zhiying Xu, Zhen Jia, Yida Wang, Gennady Pekhimenko
- MLSys 2025
- Readers: Everyone

View all 16 publications

Co-Authors

View all 53 co-authors