🚀🚀🚀 KernelBench使用大模型生成高效GPU内核 🚀🚀🚀
🌲 包含250个神经网络任务,涵盖了基础Kernel操作、简单的fusion pattern、完整的神经网络结构等;
🌳 使用大模型编写custom kernels替代PyTorch中的标准实现,并评估其正确性和性能提升
⭐ 技术博客:http://t.cn/A6mSEndL
🌞 代码repo:http://t.cn/A6mSEnd2
🌟 Leaderboard:http://t.cn/A6mSEndA
#机器学习##人工智能##大模型##Github#
发布于 上海
