首个AI CUDA工程师出现

leona

科技领域又有新突破。由Transformer论文的八位创始人之一Llion Jones创立的SakanaAI近日发布了全球首个「AI CUDA工程师」。

这一智能体，能够自动将PyTorch代码转换为高速优化的CUDA内核，性能提升惊人，速度是PyTorch原生实现的10-100倍。AICUDA工程师不仅能自动生成优化的CUDA内核，更通过运用进化算法优化这些内核的执行效率，实现多个运算的无缝融合。伴随这一项目推出的还有一个包含超过17,000个经过验证的CUDA内核的数据集，全部代码采用CC-By-4.0许可证发布，设计旨在促进开源模型的后续训练。