CudaLLM 训练语言模型以生成高性能 CUDA 内核

该项目提供了一个完整的流水线，用于训练 LLM，以自动生成高效且正确的 CUDA 内核。通过利用 SFT 和 RL 的两阶段流程，该框架可以对基础模型进行微调，从而编写优化的 CUDA 代码。 ...