← Back to Gallery

Triton Softmax Kernel

by ryan_a23•May 2, 2026•👁 2632 views

Custom row-wise softmax in Triton with numerically-stable max-subtract. Verified against torch.softmax — bit-identical to within fp32 epsilon.

#triton#softmax#custom-kernel
Sign in to vote or run
main.py
Terminal Output
Press "Run" to execute on a real GPU.

Comments (0)

to post comments and vote
No comments yet. Be the first to share your thoughts!