Mastering Sparse CUDA Generation through Pretrained Models and Deep Reinforcement Learning
ICLROral2026
TL;DR
We propose SparseRL, a deep reinforcement learning framework that generates high-performance CUDA code for sparse matrix operations, achieving significant improvements in both correctness and execution efficiency.
Opening excerpt from the authors’ abstract. source
Read the paper
Topics
reinforcement learning