Learn AI

Triton Kernel Mastery: From tl.load to Flash Attention

Write custom GPU kernels in Triton from first principles โ€” the block programming model, masking, fused softmax, autotuning, blocked matmul, and Flash Attention with online softmax, then PyTorch and torch.compile integration. Interview-framed with quizzes and spot-the-bug challenges.

8 Modules 4-6 Hours

GPU Performance Engineering: CUDA & SYCL

How GPUs actually execute code and how to make kernels fast โ€” the execution model, memory hierarchy, coalescing, bank conflicts, occupancy, the roofline model, warp divergence and reductions, and a profiler-driven optimization workflow. The interview backbone for GPU/perf roles.

8 Modules 4-6 Hours

GEMM from Scratch: How a Matmul Reaches 90% of Peak

Optimize a GPU matrix multiply step by step โ€” naive kernel, coalescing, shared-memory tiling, register blocking, vectorized loads, tensor cores, and closing the gap to cuBLAS. The canonical 'walk me through optimizing a kernel' interview exercise, with the real speedup numbers.

8 Modules 4-6 Hours

vLLM & Distributed Kernels: Interview Prep

A FAANG interview bootcamp for LLM inference & distributed GPU systems โ€” PagedAttention, continuous batching, and the vLLM scheduler, then collective communication, ring vs tree all-reduce, NCCL topology, and custom fused all-reduce kernels. Conceptual Q&A, spot-the-bug, and system-design rubrics.

8 Modules 4-6 Hours

PyTorch Interview Prep: Amateur to Expert

An 8-week FAANG interview bootcamp for PyTorch โ€” tensors and autograd through torch.compile, FSDP2, and ML system design. Conceptual Q&A, live-coding solutions, spot-the-bug challenges, and design rubrics, all on verified PyTorch 2.x APIs.

8 Modules 4-6 Hours

How vLLM Scales Across GPUs

A WebGL walkthrough of vLLM V1's four parallelism dimensions โ€” tensor, pipeline, Wide EP, and disaggregated serving โ€” with animated all-reduce, pipeline bubbles, expert dispatch, and KV transfer, each grounded in the merged PRs behind it.

8 Modules 1-2 Hours

vLLM Optimization Deep Dive

Master vLLM's optimization internals through Triton kernel implementations.

6 Modules 2-3 Hours

NKI Kernel Programming

Program AWS Trainium from first tile to Flash Attention using NKI โ€” the CUDA of Neuron hardware.

8 Modules 3-4 Hours

The Attention Family: A Visual Guide

A 3Blue1Brown-style WebGL journey through attention and its variants โ€” from Q/K/V and the Tร—T score matrix to MQA, GQA, MLA, sparse, linear, and FlashAttention, each animated to show how it beats the O(Tยฒ) cost.

8 Modules 2-3 Hours

Want more content on a specific topic?

Suggest topics for GPU kernels, ML optimization, or systems programming

Get in Touch