mojo-gpu-tutorials

Hands-on tutorials for GPU programming in Mojo 🔥 | Learn kernels, threads, blocks, memory layouts, and LayoutTensor with practical examples | Write once, run on NVIDIA, AMD, or Apple Silicon

---

📚 Available Tutorials

#	Topic	MDX	Colab	Video
001	GPU Introduction & Basics	📓 MDX
002	Memory Layout & LayoutTensor	📓 MDX
003	Vector Addition	📓 MDX
004	Thread Management	📓 MDX
005	Shared Memory	📓 MDX
006	Pooling Operations	📓 MDX
007	Convolution	📓 MDX
008	MAX Graph Introduction	-
009	Matrix Multiplication	📓 MDX
010	Functional Programming in Mojo	📓 MDX
011	Warp-Level Programming	📓 MDX
012	MAX Ecosystem	-
013	Block Level Programming	-

---

🎨 Interactive Visualizations

All tutorials include embedded interactive visualizers to help you understand GPU concepts visually!

💡 Colab Users: Visualizations are embedded directly in the notebooks - no downloads needed! Just run and interact.

🖥️ MDX Users: Use VS Code's Live Preview extension to view visualizations locally. Right-click any HTML file in visualize/ folder → Show Preview.

---

🚀 Getting Started with Colab (Recommended)

1. Click any Colab badge from the tutorials table above
2. Enable GPU: Runtime → Change runtime type → T4 GPU
3. Run cells and explore embedded visualizations! 🔥

📝 Note: Mojo code in Colab must be in a single block with a main() function.

---

🏠 Running MDX Notebooks Locally (Advanced)

For a full notebook experience with code split across cells:

Setup:

git clone https://github.com/abhisheksreesaila/mojo-gpu-tutorials.git
cd mojo-gpu-tutorials
curl -fsSL https://pixi.sh/install.sh | sh  # Install Pixi
pixi add mojo                                # Install Mojo
pixi shell                                   # Activate environment
code .                                       # Open in VS Code

Install Extensions:

• Markdown Lab - for running .mdx notebooks
• Live Preview - for viewing HTML visualizations (right-click HTML files → Show Preview)

Requirements: VS Code, Linux/macOS (Windows: WSL2), GPU optional

---

🙏 Acknowledgements

The code examples and implementations in these tutorials are adapted from Modular's GPU Puzzles and the official Mojo documentation. These materials have been reorganized and enhanced with detailed explanations to make GPU programming concepts more accessible to learners.

Special thanks to the Modular team for their excellent documentation and examples.

---

💬 Need Help?

• 📖 Mojo Notebook Setup
• 🐛 Mojo Installation Instructions
• 💡 Google Colab Guide

---

Happy GPU Programming! 🔥🚀