OAI Triton 是 OpenAI 开发的一种类似 Python 的开源编程语言，主要用于简化 GPU 代码的编写，特别针对深度学习和神经网络的高性能计算任务。Triton 可以让没有 CUDA 编程经验的研究人员和工程师也能高效地编写 GPU 代码，性能在很多情况下能媲美专家级代码。其设计目标是简化复杂 GPU 编程的流程，通过自动执行数据移动和计算优化等，使得开发者能够轻松达到或超过使用 CUDA 直接编写代码的效率.
主要特点
- 类 Python 语法，易学易用。
- 减少 GPU 编程的复杂性，特别是针对神经网络的矩阵运算。
- 自动优化内存传输（DRAM 到 SRAM）和计算分配。
- 支持高性能并行计算，能够灵活分配计算任务到多个 CUDA 核心。
- 目前支持 Nvidia GPU，不支持 AMD GPU 或 CPU 编译。
Triton 的定位
Triton 旨在成为深度学习中 CUDA 的可行替代方案，让研究人员和开发者能够在减少编写代码复杂度的同时，达到或超过传统 CUDA 内核的性能.
简言之，OAI Triton 是一门专门为 GPU 编程设计的，类 Python 的开源语言，用于高效开发深度学习计算内核。