介绍Triton是OpenAI开发的一种编程语言，帮助没有CUDA经验的开发者快速编写高性能GPU算子，实现加速。 作为上手的第一次尝试，本文以Softmax为例，体验一下triton的编写流程。Softmax是深度学习中一个基础算子，可用来将原始分数（Logits）转化为概率分布，常用于分类网络。 一维场景：$X,Y \in \mathbb{R}^{N}$ \text{Softmax}(...