Изучаем Triton по одному ядру за раз: Softmax - https://ideipro.ru/izuchaem-triton-po-odnomu-yadru-za-raz-softmax/