2 papers
cs.LG2025
FastForward Pruning: Efficient LLM Pruning via Single-Step Reinforcement Learning
Xin Yuan, Siqi Li, Jiateng Wei +7
Pruning is an effective method for compressing Large Language Models, but finding an optimal, non-uniform layer-wise sparsity allocation remains a key challenge. While heuristic me…
cs.LG2025
ConCuR: Conciseness Makes State-of-the-Art Kernel Generation
Lingcheng Kong, Jiateng Wei, Hanzhang Shen +1
GPU kernel generation by LLMs has recently experienced rapid development, leveraging test-time scaling and reinforcement learning techniques. However, a key challenge for kernel ge…