Solve puzzles. Learn CUDA.
☆11,997Sep 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,985Jul 15, 2024Updated last year
- What would you do with 1000 H100s...☆1,161Jan 10, 2024Updated 2 years ago
- ☆498Oct 18, 2024Updated last year
- Puzzles for learning Triton☆2,336Updated this week
- Puzzles for exploring transformers☆387May 4, 2023Updated 2 years ago
- Development repository for the Triton language and compiler☆18,656Mar 14, 2026Updated last week
- Machine Learning Engineering Open Book☆17,440Updated this week
- Material for gpu-mode lectures☆5,841Feb 1, 2026Updated last month
- LLM training in simple, raw C/CUDA☆29,143Jun 26, 2025Updated 8 months ago
- You like pytorch? You like micrograd? You love tinygrad! ❤️☆31,592Updated this week
- CUDA Templates and Python DSLs for High-Performance Linear Algebra☆9,442Updated this week
- GPU programming related news and material links☆2,047Mar 8, 2026Updated last week
- Fast and memory-efficient exact attention☆22,832Updated this week
- A playbook for systematically maximizing the performance of deep learning models.☆29,918Jun 18, 2024Updated last year
- Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more☆35,108Updated this week
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆9,430Feb 20, 2026Updated last month
- A PyTorch native platform for training generative AI models☆5,162Updated this week
- Tile primitives for speedy kernels☆3,232Updated this week
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆55,030Nov 12, 2025Updated 4 months ago
- A minimal GPU design in Verilog to learn how GPUs work from the ground up☆11,969Aug 18, 2024Updated last year
- A high-throughput and memory-efficient inference and serving engine for LLMs☆73,479Updated this week
- DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.☆41,807Mar 13, 2026Updated last week
- Simple and efficient pytorch-native transformer text generation in <1000 LOC of python.☆6,187Aug 22, 2025Updated 6 months ago
- Inference Llama 2 in one file of pure C☆19,262Aug 6, 2024Updated last year
- A minimal PyTorch re-implementation of the OpenAI GPT (Generative Pretrained Transformer) training☆23,919Aug 15, 2024Updated last year
- Efficient Triton Kernels for LLM Training☆6,216Updated this week
- LLM inference in C/C++☆98,098Updated this week
- A puzzle to learn about prompting☆136May 12, 2023Updated 2 years ago
- llama3 implementation one matrix multiplication at a time☆15,252May 23, 2024Updated last year
- 📚LeetCUDA: Modern CUDA Learn Notes with PyTorch for Beginners🐑, 200+ CUDA Kernels, Tensor Cores, HGEMM, FA-2 MMA.🎉☆9,932Updated this week
- The full minitorch student suite.☆2,320Aug 17, 2024Updated last year
- NumPy & SciPy for GPU☆10,847Mar 14, 2026Updated last week
- Tensor library for machine learning☆14,252Updated this week
- Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.☆30,926Mar 10, 2026Updated last week
- LLM101n: Let's build a Storyteller☆36,489Aug 1, 2024Updated last year
- Hackable and optimized Transformers building blocks, supporting a composable construction.☆10,373Updated this week
- A tiny scalar-valued autograd engine and a neural net library on top of it with PyTorch-like API☆15,046Aug 8, 2024Updated last year
- Accessible large language models via k-bit quantization for PyTorch.☆8,052Updated this week
- AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (N…☆4,709Updated this week