Alternatives and similar repositories for Tensor-Puzzles

Users that are interested in Tensor-Puzzles are comparing it to the libraries listed below

• srush / GPU-Puzzles
  Solve puzzles. Learn CUDA.
  ☆11,206Updated 10 months ago
• srush / LLM-Training-Puzzles
  What would you do with 1000 H100s...
  ☆1,056Updated last year
• srush / Triton-Puzzles
  Puzzles for learning Triton
  ☆1,726Updated 7 months ago
• google-deepmind / penzai
  A JAX research toolkit for building, editing, and visualizing neural networks.
  ☆1,795Updated last week
• pytorch-labs / gpt-fast
  Simple and efficient pytorch-native transformer text generation in <1000 LOC of python.
  ☆6,000Updated 2 months ago
• minitorch / minitorch
  The full minitorch student suite.
  ☆2,121Updated 10 months ago
• srush / Autodiff-Puzzles
  ☆439Updated 8 months ago
• facebookresearch / schedule_free
  Schedule-Free Optimization in PyTorch
  ☆2,185Updated last month
• pytorch / torchtitan
  A PyTorch native platform for training generative AI models
  ☆3,977Updated this week
• gpu-mode / resource-stream
  GPU programming related news and material links
  ☆1,604Updated 5 months ago
• srush / Transformer-Puzzles
  Puzzles for exploring transformers
  ☆353Updated 2 years ago
• HazyResearch / ThunderKittens
  Tile primitives for speedy kernels
  ☆2,489Updated last week
• stas00 / the-art-of-debugging
  The Art of Debugging
  ☆895Updated 10 months ago
• gpu-mode / lectures
  Material for gpu-mode lectures
  ☆4,661Updated 2 weeks ago
• tysam-code / hlb-CIFAR10
  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
• Lightning-AI / lightning-thunder
  Thunder gives you PyTorch models superpowers for training and inference. Unlock out-of-the-box optimizations for performance, memory and …
  ☆1,369Updated this week
• xl0 / lovely-tensors
  Tensors, for human consumption
  ☆1,261Updated last week
• clu0 / unet.cu
  UNet diffusion model in pure CUDA
  ☆609Updated last year
• explosion / curated-transformers
  🤖 A PyTorch library of curated Transformer models and their composable components
  ☆892Updated last year
• jacobhilton / deep_learning_curriculum
  Language model alignment-focused deep learning curriculum
  ☆1,412Updated 10 months ago
• HazyResearch / aisys-building-blocks
  Building blocks for foundation models.
  ☆510Updated last year
• EleutherAI / cookbook
  Deep learning for dummies. All the practical details and useful utilities that go into working with real models.
  ☆802Updated 3 weeks ago
• stas00 / ml-engineering
  Machine Learning Engineering Open Book
  ☆14,126Updated this week
• huggingface / picotron
  Minimalistic 4D-parallelism distributed training framework for education purpose
  ☆1,561Updated last month
• pytorch / tensordict
  TensorDict is a pytorch dedicated tensor container.
  ☆931Updated this week
• JonasGeiping / cramming
  Cramming the training of a (BERT-type) language model into limited compute.
  ☆1,336Updated last year
• KellerJordan / modded-nanogpt
  NanoGPT (124M) in 3 minutes
  ☆2,721Updated last week
• pytorch / ao
  PyTorch native quantization and sparsity for training and inference
  ☆2,138Updated this week
• AI-Hypercomputer / maxtext
  A simple, performant and scalable Jax LLM!
  ☆1,803Updated this week
• karpathy / nano-llama31
  nanoGPT style version of Llama 3.1
  ☆1,389Updated 10 months ago