srush / Tensor-PuzzlesLinks
Solve puzzles. Improve your pytorch.
☆3,609Updated 11 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,180Updated 9 months ago
- The full minitorch student suite.☆2,121Updated 10 months ago
- Puzzles for learning Triton☆1,726Updated 7 months ago
- What would you do with 1000 H100s...☆1,055Updated last year
- ☆436Updated 8 months ago
- A JAX research toolkit for building, editing, and visualizing neural networks.☆1,796Updated this week
- NanoGPT (124M) in 3 minutes☆2,699Updated last week
- Schedule-Free Optimization in PyTorch☆2,180Updated last month
- Machine Learning Engineering Open Book☆14,082Updated 2 weeks ago
- GPU programming related news and material links☆1,594Updated 5 months ago
- An autoregressive character-level language model for making more things☆3,139Updated last year
- Train to 94% on CIFAR-10 in <6.3 seconds on a single A100. Or ~95.79% in ~110 seconds (or less!)☆1,271Updated 6 months ago
- Puzzles for exploring transformers☆350Updated 2 years ago
- Deep learning for dummies. All the practical details and useful utilities that go into working with real models.☆801Updated 2 weeks ago
- Material for gpu-mode lectures☆4,636Updated last week
- Minimalistic 4D-parallelism distributed training framework for education purpose☆1,554Updated 3 weeks ago
- A tiny scalar-valued autograd engine and a neural net library on top of it with PyTorch-like API☆12,173Updated 10 months ago
- Video+code lecture on building nanoGPT from scratch☆4,168Updated 10 months ago
- The n-gram Language Model☆1,428Updated 10 months ago
- Package for extracting and mapping the results of every single tensor operation in a PyTorch model in one line of code.☆594Updated 3 months ago
- A PyTorch native platform for training generative AI models☆3,953Updated this week
- UNet diffusion model in pure CUDA☆608Updated 11 months ago
- Minimal, clean code for the Byte Pair Encoding (BPE) algorithm commonly used in LLM tokenization.☆9,717Updated 11 months ago
- Simple and efficient pytorch-native transformer text generation in <1000 LOC of python.☆5,998Updated 2 months ago
- The Art of Debugging☆893Updated 10 months ago
- Implementing DeepSeek R1's GRPO algorithm from scratch☆1,438Updated 2 months ago
- 🧠 A study guide to learn about Transformers☆1,595Updated 2 years ago
- An ML Systems Onboarding list☆818Updated 5 months ago
- Notebooks and various random fun☆1,095Updated 2 years ago
- Tile primitives for speedy kernels☆2,478Updated this week