Triton-Puzzles and Tensor-Puzzles
These are complementary learning resources—Triton-Puzzles teaches the Triton compiler for GPU kernels while Tensor-Puzzles teaches PyTorch fundamentals, so learners typically progress from PyTorch understanding to optimized Triton kernel implementation.
About Triton-Puzzles
gpu-mode/Triton-Puzzles
Puzzles for learning Triton
This project offers interactive puzzles to help deep learning practitioners learn Triton, an open-source language for programming AI accelerators. You'll solve challenges that involve memory operations, starting with simple examples and progressing to advanced algorithms like Flash Attention. This is for AI/ML engineers and researchers who want to optimize their models for GPUs using Triton.
About Tensor-Puzzles
srush/Tensor-Puzzles
Solve puzzles. Improve your pytorch.
This project offers a collection of 21 tensor programming puzzles designed to sharpen your skills in PyTorch or NumPy. Each puzzle challenges you to re-implement a standard library function using only fundamental operations like broadcasting, arithmetic, and indexing. It's ideal for data scientists, machine learning engineers, or anyone working with numerical data who wants to deepen their understanding of tensor manipulation.
Scores updated daily from GitHub, PyPI, and npm data. How scores work