srush / Tensor-PuzzlesLinks
Solve puzzles. Improve your pytorch.
☆3,912Updated 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,932Updated last year
- The full minitorch student suite.☆2,292Updated last year
- Puzzles for learning Triton☆2,283Updated last year
- What would you do with 1000 H100s...☆1,151Updated 2 years ago
- GPU programming related news and material links☆1,955Updated 4 months ago
- A JAX research toolkit for building, editing, and visualizing neural networks.☆1,860Updated 7 months ago
- ☆492Updated last year
- Material for gpu-mode lectures☆5,679Updated last week
- Schedule-Free Optimization in PyTorch☆2,254Updated 8 months ago
- The Art of Debugging Open Book☆1,276Updated 3 weeks ago
- Train to 94% on CIFAR-10 in <6.3 seconds on a single A100. Or ~95.79% in ~110 seconds (or less!)☆1,299Updated last year
- Simple and efficient pytorch-native transformer text generation in <1000 LOC of python.☆6,180Updated 5 months ago
- Puzzles for exploring transformers☆384Updated 2 years ago
- NanoGPT (124M) in 2 minutes☆4,589Updated last week
- Tile primitives for speedy kernels☆3,120Updated this week
- Language model alignment-focused deep learning curriculum☆1,531Updated last year
- A PyTorch native platform for training generative AI models☆5,023Updated last week
- Minimalistic 4D-parallelism distributed training framework for education purpose☆2,058Updated 5 months ago
- UNet diffusion model in pure CUDA☆661Updated last year
- Deep learning for dummies. All the practical details and useful utilities that go into working with real models.☆829Updated 6 months ago
- Tensors, for human consumption☆1,353Updated 2 weeks ago
- Minimal, clean code for the Byte Pair Encoding (BPE) algorithm commonly used in LLM tokenization.☆10,293Updated last year
- PyTorch compiler that accelerates training and inference. Get built-in optimizations for performance, memory, parallelism, and easily wri…☆1,439Updated this week
- Pen and paper exercises in machine learning☆2,577Updated last year
- ☆4,112Updated last year
- Flax is a neural network library for JAX that is designed for flexibility.☆7,058Updated last week
- An ML Systems Onboarding list☆981Updated last year
- An autoregressive character-level language model for making more things☆3,635Updated last year
- JAX - A curated list of resources https://github.com/google/jax☆2,036Updated 2 weeks ago
- PyTorch native post-training library☆5,660Updated this week