跳到正文
原文
OpenAI News·· 2021-07-28AI 评分55

OpenAI 发布 Triton 1.0:面向 GPU 的高性能深度学习编程框架

Introducing Triton: Open-source GPU programming for neural networks

AI 导读

OpenAI 正式发布 Triton 1.0,这是一个用于编写高性能深度学习原语的语言和编译器。Triton 允许开发者使用类似 Python 的语法直接操作 GPU,无需掌握 CUDA C++ 或复杂的底层优化技巧,即可实现接近专家手写代码的性能。该工具旨在简化自定义神经网络层、注意力机制及优化器的开发流程,目前已在 GitHub 开源并支持 PyTorch 集成。

原文

We’re releasing Triton 1.0, an open-source Python-like programming language which enables researchers with no CUDA experience to write highly efficient GPU code—most of the time on par with what an expert would be able to produce.

来源:OpenAI News · openai.com