srush / Tensor-Puzzles
Solve puzzles. Improve your pytorch.
☆3,486Updated 8 months ago
Alternatives and similar repositories for Tensor-Puzzles:
Users that are interested in Tensor-Puzzles are comparing it to the libraries listed below
- Solve puzzles. Learn CUDA.☆10,751Updated 6 months ago
- Puzzles for learning Triton☆1,527Updated 4 months ago
- What would you do with 1000 H100s...☆1,021Updated last year
- The full minitorch student suite.☆2,034Updated 7 months ago
- ☆423Updated 5 months ago
- GPU programming related news and material links☆1,421Updated 2 months ago
- Puzzles for exploring transformers☆333Updated last year
- Schedule-Free Optimization in PyTorch☆2,116Updated 3 weeks ago
- A JAX research toolkit for building, editing, and visualizing neural networks.☆1,746Updated 3 months ago
- Tile primitives for speedy kernels☆2,170Updated this week
- A PyTorch native library for large model training☆3,488Updated this week
- UNet diffusion model in pure CUDA☆600Updated 8 months ago
- Simple and efficient pytorch-native transformer text generation in <1000 LOC of python.☆5,892Updated last week
- An ML Systems Onboarding list☆734Updated 2 months ago
- Train to 94% on CIFAR-10 in <6.3 seconds on a single A100. Or ~95.79% in ~110 seconds (or less!)☆1,252Updated 3 months ago
- NanoGPT (124M) in 3 minutes☆2,417Updated last week
- Tensors, for human consumption☆1,202Updated 4 months ago
- Minimalistic 4D-parallelism distributed training framework for education purpose☆948Updated 2 weeks ago
- An autoregressive character-level language model for making more things☆2,948Updated 9 months ago
- If tinygrad wasn't small enough for you...☆699Updated last year
- Building blocks for foundation models.☆467Updated last year
- Language model alignment-focused deep learning curriculum☆1,345Updated 7 months ago
- The Art of Debugging☆861Updated 7 months ago
- Cramming the training of a (BERT-type) language model into limited compute.☆1,325Updated 9 months ago
- TensorDict is a pytorch dedicated tensor container.☆899Updated this week
- A library for mechanistic interpretability of GPT-style language models☆1,975Updated this week
- Deep learning for dummies. All the practical details and useful utilities that go into working with real models.☆783Updated 3 weeks ago
- The purpose of this repo is to make it easy to get started with JAX, Flax, and Haiku. It contains my "Machine Learning with JAX" series o…☆705Updated last year
- A tiny scalar-valued autograd engine and a neural net library on top of it with PyTorch-like API☆11,453Updated 7 months ago
- Llama from scratch, or How to implement a paper without crying☆550Updated 9 months ago