Zheren Dong

I’m a research engineer on the AI Research Team at Augment Code, where I focus on post-training LLMs and data curation strategies that improve retrieval performance for coding agents in production. This involves building pipelines to process real user data, handle distribution shifts, and maintain training data quality. Outside of work, I pursue independent ML research; my recent work on spelling-aware embeddings shows how simple architectural changes can improve language modeling across benchmarks.

Previously, I worked at Applied Intuition and Rivos (now part of Meta).