#411 · Primary category: Education & Research

Triton-Puzzles

machine-learning puzzle

Puzzles for learning Triton

Project last updated:04/01/26

GitHub Stars

2.6K

Forks

252

Contributors

11

License

Apache-2.0

Why we included this project

Triton sits between high-level PyTorch-like code and raw CUDA, and it powers a lot of modern deep learning. Most tutorials expect you to already understand CUDA's memory model, but this puzzle set starts from the basics and uses Triton's interpreter, so you can work through everything in a Colab notebook without a GPU. You begin with simple tensor ops and work up to Flash Attention and quantized networks, which is a solid way to get a feel for low-level performance thinking. That makes it useful for anyone writing custom kernels or trying to read performance-sensitive code, and also one of the gentler introductions to accelerator programming around.

Articles for this project

No articles for this project yet.

To suggest a topic or contribute an article, contact us.

Related projects in this category