Solve puzzles. Improve your pytorch.
☆3,950Jul 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,959Sep 1, 2024Updated last year
- ☆497Oct 18, 2024Updated last year
- Puzzles for exploring transformers☆386May 4, 2023Updated 2 years ago
- What would you do with 1000 H100s...☆1,154Jan 10, 2024Updated 2 years ago
- Puzzles for learning Triton☆2,314Nov 18, 2024Updated last year
- A puzzle to learn about prompting☆135May 12, 2023Updated 2 years ago
- Machine Learning Engineering Open Book☆17,162Feb 21, 2026Updated last week
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆9,401Feb 20, 2026Updated last week
- You like pytorch? You like micrograd? You love tinygrad! ❤️☆31,424Updated 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
- Development repository for the Triton language and compiler☆18,460Feb 22, 2026Updated last week
- Simple and efficient pytorch-native transformer text generation in <1000 LOC of python.☆6,184Aug 22, 2025Updated 6 months ago
- Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more☆34,940Updated this week
- The full minitorch student suite.☆2,307Aug 17, 2024Updated last year
- A JAX research toolkit for building, editing, and visualizing neural networks.☆1,869Jun 22, 2025Updated 8 months ago
- Deep learning for dummies. All the practical details and useful utilities that go into working with real models.☆831Updated this week
- A PyTorch native platform for training generative AI models☆5,098Updated this week
- Annotated version of the Mamba paper☆497Feb 27, 2024Updated 2 years ago
- 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
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆53,743Nov 12, 2025Updated 3 months ago
- Tile primitives for speedy kernels☆3,183Updated this week
- LLM training in simple, raw C/CUDA☆28,940Jun 26, 2025Updated 8 months ago
- A minimal PyTorch re-implementation of the OpenAI GPT (Generative Pretrained Transformer) training☆23,667Aug 15, 2024Updated last year
- Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/☆2,792Updated this week
- The Art of Debugging Open Book☆1,299Feb 16, 2026Updated last week
- DSPy: The framework for programming—not prompting—language models☆32,381Updated this week
- Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/☆1,736Feb 16, 2026Updated last week
- Fast and memory-efficient exact attention☆22,361Updated this week
- Material for gpu-mode lectures☆5,773Feb 1, 2026Updated last month
- JAX - A curated list of resources https://github.com/google/jax☆2,058Jan 20, 2026Updated last month
- Flax is a neural network library for JAX that is designed for flexibility.☆7,082Feb 22, 2026Updated last week
- Minimal, clean code for the Byte Pair Encoding (BPE) algorithm commonly used in LLM tokenization.☆10,329Jul 1, 2024Updated last year
- Schedule-Free Optimization in PyTorch☆2,256May 21, 2025Updated 9 months ago
- Minimalistic 4D-parallelism distributed training framework for education purpose☆2,090Aug 26, 2025Updated 6 months ago
- llama3 implementation one matrix multiplication at a time☆15,243May 23, 2024Updated last year
- Tensors, for human consumption☆1,360Jan 22, 2026Updated last month
- A bibliography and survey of the papers surrounding o1☆1,212Nov 16, 2024Updated last year
- Accessible large language models via k-bit quantization for PyTorch.☆7,997Updated this week