From the 1 of 10 linked papers with an AI index.
10 papers
LEAP: Lean Environment-Feedback via Adaptive Pruning for Code RL in GPU Kernel Generation
Tankun Li, Zhi Chen, Yaohua Tang
Post-training large language models (LLMs) via reinforcement learning (RL) has significantly advanced code generation capabilities. To bypass the heavy memory footprint of critic n…
TCPO: Turn-Level Credit Policy Optimization
Sicong Liao, Zhi Chen, Yaohua Tang
Verifier-guided reinforcement learning has become a powerful paradigm for improving LLM reasoning. In multi-turn settings, models receive a verifier score after each turn and itera…
Beyond the Best Teacher: Expanding and Compressing the Reasoning Solution Manifold
Songshuo Lu, Zhi Chen, Yaohua Tang
The paper proposes an expand‑then‑compress framework that builds a diverse set of RL‑trained teacher models and then distills them into a single student model, improving reasoning,…
Latent Reasoning with Normalizing Flows
Guancheng Tu, Xiangjun Fu, Suhao Yu +5
Large language models often improve reasoning by generating explicit chain-of-thought (CoT), demonstrating the importance of intermediate computation. However, textual CoT forces t…
MusaCoder: Native GPU Kernel Generation with Full-Stack Training on Moore Threads GPU
Kun Cheng, Songshuo Lu, Sicong Liao +7
Native GPU kernel generation turns high-level tensor programs into executable, efficient low-level code. Existing Large Language Models (LLMs) struggle with this task, while execut…
LiteGUI: Distilling Compact GUI Agents with Reinforcement Learning
Yubin Wu, Zicheng Cai, Liping Ning +4
Developing lightweight, on-device vision-language GUI agents is essential for efficient cross-platform automated interaction. However, current on-device agents are constrained by l…