JasonDai大数据AI
24-12-05 18:12 微博认证:AI博主

🚀🚀🚀 KernelBench使用大模型生成高效GPU内核 🚀🚀🚀
🌲 包含250个神经网络任务,涵盖了基础Kernel操作、简单的fusion pattern、完整的神经网络结构等;
🌳 使用大模型编写custom kernels替代PyTorch中的标准实现,并评估其正确性和性能提升

⭐ 技术博客:http://t.cn/A6mSEndL
🌞 代码repo:http://t.cn/A6mSEnd2
🌟 Leaderboard:http://t.cn/A6mSEndA

#机器学习##人工智能##大模型##Github#

发布于 上海