srush / Tensor-PuzzlesLinks
Solve puzzles. Improve your pytorch.
☆3,579Updated 10 months ago
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,053Updated 9 months ago
- What would you do with 1000 H100s...☆1,050Updated last year
- Puzzles for learning Triton☆1,671Updated 6 months ago
- The full minitorch student suite.☆2,087Updated 9 months ago
- GPU programming related news and material links☆1,540Updated 5 months ago
- A JAX research toolkit for building, editing, and visualizing neural networks.☆1,784Updated last month
- ☆433Updated 7 months ago
- Minimalistic 4D-parallelism distributed training framework for education purpose☆1,518Updated this week
- Pen and paper exercises in machine learning☆2,353Updated last year
- Puzzles for exploring transformers☆348Updated 2 years ago
- NanoGPT (124M) in 3 minutes☆2,610Updated last week
- Tile primitives for speedy kernels☆2,420Updated this week
- Llama from scratch, or How to implement a paper without crying☆567Updated last year
- An autoregressive character-level language model for making more things☆3,109Updated last year
- A PyTorch native platform for training generative AI models☆3,891Updated this week
- Machine Learning Engineering Open Book☆13,933Updated this week
- Train to 94% on CIFAR-10 in <6.3 seconds on a single A100. Or ~95.79% in ~110 seconds (or less!)☆1,265Updated 5 months ago
- Schedule-Free Optimization in PyTorch☆2,162Updated 2 weeks ago
- An ML Systems Onboarding list☆794Updated 4 months ago
- The n-gram Language Model☆1,421Updated 10 months ago
- UNet diffusion model in pure CUDA☆606Updated 11 months ago
- ☆4,082Updated last year
- A library for mechanistic interpretability of GPT-style language models☆2,217Updated last week
- Minimal, clean code for the Byte Pair Encoding (BPE) algorithm commonly used in LLM tokenization.☆9,671Updated 11 months ago
- Simple and efficient pytorch-native transformer text generation in <1000 LOC of python.☆5,975Updated last month
- Language model alignment-focused deep learning curriculum☆1,397Updated 9 months ago
- Tensors, for human consumption☆1,252Updated 2 weeks ago
- The Autograd Engine☆609Updated 8 months ago
- Material for gpu-mode lectures☆4,530Updated 3 months ago
- Implementation of papers in 100 lines of code.☆1,546Updated 3 weeks ago