Solve puzzles. Improve your pytorch.
☆3,985Jul 15, 2024Updated last year
Alternatives and similar repositories for Tensor-Puzzles
Users that are interested in Tensor-Puzzles are comparing it to the libraries listed below
Sorting:
- Solve puzzles. Learn CUDA.☆11,997Sep 1, 2024Updated last year
- ☆498Oct 18, 2024Updated last year
- Puzzles for exploring transformers☆387May 4, 2023Updated 2 years ago
- What would you do with 1000 H100s...☆1,161Jan 10, 2024Updated 2 years ago
- Puzzles for learning Triton☆2,336Updated this week
- A puzzle to learn about prompting☆136May 12, 2023Updated 2 years ago
- Machine Learning Engineering Open Book☆17,440Updated this week
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆9,430Feb 20, 2026Updated last month
- You like pytorch? You like micrograd? You love tinygrad! ❤️☆31,592Updated this week
- GPU programming related news and material links☆2,047Mar 8, 2026Updated last week
- A playbook for systematically maximizing the performance of deep learning models.☆29,918Jun 18, 2024Updated last year
- The full minitorch student suite.☆2,320Aug 17, 2024Updated last year
- Simple and efficient pytorch-native transformer text generation in <1000 LOC of python.☆6,187Aug 22, 2025Updated 6 months ago
- Development repository for the Triton language and compiler☆18,708Updated this week
- Deep learning for dummies. All the practical details and useful utilities that go into working with real models.☆836Updated this week
- Annotated version of the Mamba paper☆499Feb 27, 2024Updated 2 years ago
- A JAX research toolkit for building, editing, and visualizing neural networks.☆1,873Jun 22, 2025Updated 9 months ago
- Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more☆35,108Updated this week
- A PyTorch native platform for training generative AI models☆5,162Updated this week
- An interactive exploration of Transformer programming.☆274Nov 15, 2023Updated 2 years ago
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆55,030Nov 12, 2025Updated 4 months ago
- LLM training in simple, raw C/CUDA☆29,216Jun 26, 2025Updated 8 months ago
- 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
- Tile primitives for speedy kernels☆3,232Updated this week
- A minimal PyTorch re-implementation of the OpenAI GPT (Generative Pretrained Transformer) training☆23,950Aug 15, 2024Updated last year
- DSPy: The framework for programming—not prompting—language models☆32,853Updated this week
- Material for gpu-mode lectures☆5,841Feb 1, 2026Updated last month
- Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/☆1,762Feb 16, 2026Updated last month
- Fast and memory-efficient exact attention☆22,832Updated this week
- The Art of Debugging Open Book☆1,326Updated this week
- Introductory lecture on Pytorch☆17May 8, 2022Updated 3 years ago
- A bibliography and survey of the papers surrounding o1☆1,212Nov 16, 2024Updated last year
- Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/☆2,821Mar 9, 2026Updated last week
- An ML Systems Onboarding list☆1,008Feb 19, 2026Updated last month
- llama3 implementation one matrix multiplication at a time☆15,252May 23, 2024Updated last year
- JAX - A curated list of resources https://github.com/google/jax☆2,074Jan 20, 2026Updated 2 months ago
- LLM101n: Let's build a Storyteller☆36,559Aug 1, 2024Updated last year
- Minimalistic 4D-parallelism distributed training framework for education purpose☆2,116Aug 26, 2025Updated 6 months ago
- Schedule-Free Optimization in PyTorch☆2,265May 21, 2025Updated 10 months ago