2. Getting Started
  3. 🔥 Introduction
  4. 🧭 Puzzles Usage Guide
  6. Part I: GPU Fundamentals
  7. Puzzle 1: Map
    1. 🔰 Raw Memory Approach
    2. 💡 Preview: Modern Approach with LayoutTensor
  9. Puzzle 2: Zip
  10. Puzzle 3: Guards
  11. Puzzle 4: 2D Map
    1. 🔰 Raw Memory Approach
    2. 📚 Learn about LayoutTensor
    3. 🚀 Modern 2D Operations
  13. Puzzle 5: Broadcast
    1. 🔰 Raw Memory Approach
    2. 📐 LayoutTensor Version
  15. Puzzle 6: Blocks
  16. Puzzle 7: 2D Blocks
    1. 🔰 Raw Memory Approach
    2. 📐 LayoutTensor Version
  18. Puzzle 8: Shared Memory
    1. 🔰 Raw Memory Approach
    2. 📐 LayoutTensor Version
  21. Part II: 🧮 GPU Algorithms
  22. Puzzle 9: Pooling
    1. 🔰 Raw Memory Approach
    2. 📐 LayoutTensor Version
  24. Puzzle 10: Dot Product
    1. 🔰 Raw Memory Approach
    2. 📐 LayoutTensor Version
  26. Puzzle 11: 1D Convolution
    1. 🔰 Simple Version
    2. ⭐ Block Boundary Version
  28. Puzzle 12: Prefix Sum
    1. 🔰 Simple Version
    2. ⭐ Complete Version
  30. Puzzle 13: Axis Sum
  31. Puzzle 14: Matrix Multiplication (MatMul)
    1. 🔰 Naïve Version with Global Memory
    2. 📚 Learn about Roofline Model
    3. 🤝 Shared Memory Version
    4. 📐 Tiled Version
  34. Part III: 🐍 Interfacing with Python via MAX Graph Custom Ops
  35. Puzzle 15: 1D Convolution Op
  36. Puzzle 16: Softmax Op
  37. Puzzle 17: Attention Op
  38. 🎯 Bonus Challenges
  40. Part IV: 🔥 PyTorch Custom Ops Integration
  41. Puzzle 18: PyTorch Custom Op Basics
  42. Puzzle 19: Integration with torch.compile
  44. Part V: 🌊 Mojo Functional Patterns and Benchmarking
  45. Puzzle 20: GPU Functional Programming Patterns
    1. elementwise - Basic GPU Functional Operations
    2. tile - Memory-Efficient Tiled Processing
    3. Vectorization - SIMD Control
    4. 🧠 GPU Threading vs SIMD Concepts
    5. 📊 Benchmarking in Mojo
  48. Part VI: ⚡ Warp-Level Programming
  49. Puzzle 21: Warp Fundamentals
    1. Warp lanes & SIMT execution
    2. warp.sum() Essentials
    3. 📊 When to Use Warp Programming
  51. Puzzle 22: Essential Warp Operations
    1. 🔄 warp.shuffle_down() Communication
    2. 🔀 warp.shuffle_xor() Butterfly Patterns
    3. 📡 warp.broadcast() Distribution
  53. Puzzle 23: Advanced Warp Patterns
    1. 🧮 warp.prefix_sum() Scan Operations
    2. lane_group_* Sub-warp Operations
    3. Combining with Functional Patterns
  55. 📋 Quick Reference: Essential Warp Operations
  57. Part VII: 🧠 Advanced Memory Operations
  58. Puzzle 24: Memory Coalescing
    1. 📚 Understanding Coalesced Access
    2. Optimized Access Patterns
    3. 🔧 Troubleshooting Memory Issues
  60. Puzzle 25: Async Memory Operations
  61. Puzzle 26: Memory Fences & Atomics
  62. Puzzle 27: Prefetching & Caching
  64. Part VIII: 📊 Performance Analysis & Optimization
  65. Puzzle 28: GPU Profiling Basics
  66. Puzzle 29: Occupancy Optimization
  67. Puzzle 30: Bank Conflicts
    1. 📚 Understanding Shared Memory Banks
    2. Conflict-Free Patterns
  70. Part IX: 🚀 Advanced GPU Features
  71. Puzzle 31: Tensor Core Operations
  72. Puzzle 32: Random Number Generation
  73. Puzzle 33: Advanced Synchronization
  75. Part X: 🌐 Multi-GPU & Advanced Applications
  76. Puzzle 34: Multi-Stream Programming
  77. Puzzle 35: Multi-GPU Basics
  78. Puzzle 36: End-to-End Optimization Case Study
  79. 🎯 Advanced Bonus Challenges