srush / Tensor-PuzzlesLinks
Solve puzzles. Improve your pytorch.
☆3,616Updated 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,206Updated 10 months ago
- What would you do with 1000 H100s...☆1,056Updated last year
- Puzzles for learning Triton☆1,726Updated 7 months ago
- A JAX research toolkit for building, editing, and visualizing neural networks.☆1,795Updated last week
- Simple and efficient pytorch-native transformer text generation in <1000 LOC of python.☆6,000Updated 2 months ago
- The full minitorch student suite.☆2,121Updated 10 months ago
- ☆439Updated 8 months ago
- Schedule-Free Optimization in PyTorch☆2,185Updated last month
- A PyTorch native platform for training generative AI models☆3,977Updated this week
- GPU programming related news and material links☆1,604Updated 5 months ago
- Puzzles for exploring transformers☆353Updated 2 years ago
- Tile primitives for speedy kernels☆2,489Updated last week
- The Art of Debugging☆895Updated 10 months ago
- Material for gpu-mode lectures☆4,661Updated 2 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,271Updated 6 months ago
- Thunder gives you PyTorch models superpowers for training and inference. Unlock out-of-the-box optimizations for performance, memory and …☆1,369Updated this week
- Tensors, for human consumption☆1,261Updated last week
- UNet diffusion model in pure CUDA☆609Updated last year
- 🤖 A PyTorch library of curated Transformer models and their composable components☆892Updated last year
- Language model alignment-focused deep learning curriculum☆1,412Updated 10 months ago
- Building blocks for foundation models.☆510Updated last year
- Deep learning for dummies. All the practical details and useful utilities that go into working with real models.☆802Updated 3 weeks ago
- Machine Learning Engineering Open Book☆14,126Updated this week
- Minimalistic 4D-parallelism distributed training framework for education purpose☆1,561Updated last month
- TensorDict is a pytorch dedicated tensor container.