Solve puzzles. Learn CUDA.
☆11,959Sep 1, 2024Updated last year
Alternatives and similar repositories for GPU-Puzzles
Users that are interested in GPU-Puzzles are comparing it to the libraries listed below
Sorting:
- Solve puzzles. Improve your pytorch.☆3,950Jul 15, 2024Updated last year
- What would you do with 1000 H100s...☆1,154Jan 10, 2024Updated 2 years ago
- Puzzles for learning Triton☆2,314Nov 18, 2024Updated last year
- ☆497Oct 18, 2024Updated last year
- Development repository for the Triton language and compiler☆18,460Feb 22, 2026Updated last week
- Machine Learning Engineering Open Book☆17,162Feb 21, 2026Updated last week
- LLM training in simple, raw C/CUDA☆28,940Jun 26, 2025Updated 8 months ago
- Puzzles for exploring transformers☆386May 4, 2023Updated 2 years ago
- Material for gpu-mode lectures☆5,773Feb 1, 2026Updated last month
- You like pytorch? You like micrograd? You love tinygrad! ❤️☆31,424Updated this week
- CUDA Templates and Python DSLs for High-Performance Linear Algebra☆9,315Updated this week
- Fast and memory-efficient exact attention☆22,361Updated this week
- Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more☆34,940Updated this week
- A playbook for systematically maximizing the performance of deep learning models.☆29,861Jun 18, 2024Updated last year
- GPU programming related news and material links☆1,997Sep 17, 2025Updated 5 months ago
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆9,401Feb 20, 2026Updated last week
- A PyTorch native platform for training generative AI models☆5,098Updated this week
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆53,743Nov 12, 2025Updated 3 months ago
- A minimal GPU design in Verilog to learn how GPUs work from the ground up☆11,766Aug 18, 2024Updated last year
- DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.☆41,648Updated this week
- A high-throughput and memory-efficient inference and serving engine for LLMs☆71,234Updated this week
- Tile primitives for speedy kernels☆3,183Updated this week
- Inference Llama 2 in one file of pure C☆19,186Aug 6, 2024Updated last year
- A minimal PyTorch re-implementation of the OpenAI GPT (Generative Pretrained Transformer) training☆23,667Aug 15, 2024Updated last year
- Simple and efficient pytorch-native transformer text generation in <1000 LOC of python.☆6,184Aug 22, 2025Updated 6 months ago
- LLM inference in C/C++☆95,726Updated this week
- Efficient Triton Kernels for LLM Training☆6,162Updated this week
- Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.☆30,860Feb 21, 2026Updated last week
- NumPy & SciPy for GPU☆10,804Updated this week
- Tensor library for machine learning☆14,152Updated this week
- llama3 implementation one matrix multiplication at a time☆15,243May 23, 2024Updated last year
- A tiny scalar-valued autograd engine and a neural net library on top of it with PyTorch-like API☆14,774Aug 8, 2024Updated last year
- AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (N…☆4,706Jan 12, 2026Updated last month
- Minimal, clean code for the Byte Pair Encoding (BPE) algorithm commonly used in LLM tokenization.☆10,329Jul 1, 2024Updated last year
- Implement a ChatGPT-like LLM in PyTorch from scratch, step by step☆86,149Feb 19, 2026Updated last week
- LLM101n: Let's build a Storyteller☆36,357Aug 1, 2024Updated last year
- Hackable and optimized Transformers building blocks, supporting a composable construction.☆10,353Feb 20, 2026Updated last week
- DSPy: The framework for programming—not prompting—language models☆32,381Updated this week
- Accessible large language models via k-bit quantization for PyTorch.☆7,997Updated this week