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.
  ☆10,919Updated 7 months ago
• minitorch / minitorch
  The full minitorch student suite.
  ☆2,055Updated 8 months ago
• srush / LLM-Training-Puzzles
  What would you do with 1000 H100s...
  ☆1,038Updated last year
• srush / Triton-Puzzles
  Puzzles for learning Triton
  ☆1,591Updated 5 months ago
• srush / Autodiff-Puzzles
  ☆428Updated 6 months ago
• google-deepmind / penzai
  A JAX research toolkit for building, editing, and visualizing neural networks.
  ☆1,759Updated this week
• facebookresearch / schedule_free
  Schedule-Free Optimization in PyTorch
  ☆2,146Updated 2 weeks ago
• gpu-mode / resource-stream
  GPU programming related news and material links
  ☆1,461Updated 3 months ago
• srush / Transformer-Puzzles
  Puzzles for exploring transformers
  ☆343Updated last year
• 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,252Updated 4 months ago
• patrick-kidger / equinox
  Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
  ☆2,324Updated this week
• karpathy / micrograd
  A tiny scalar-valued autograd engine and a neural net library on top of it with PyTorch-like API
  ☆11,713Updated 8 months ago
• clu0 / unet.cu
  UNet diffusion model in pure CUDA
  ☆602Updated 9 months ago
• xl0 / lovely-tensors
  Tensors, for human consumption
  ☆1,215Updated 5 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,328Updated this week
• pytorch / torchtitan
  A PyTorch native library for large-scale model training
  ☆3,627Updated this week
• michaelgutmann / ml-pen-and-paper-exercises
  Pen and paper exercises in machine learning
  ☆2,269Updated 11 months ago
• n2cholas / awesome-jax
  JAX - A curated list of resources https://github.com/google/jax
  ☆1,788Updated 2 months ago
• karpathy / makemore
  An autoregressive character-level language model for making more things
  ☆3,021Updated 10 months ago
• HazyResearch / ThunderKittens
  Tile primitives for speedy kernels
  ☆2,279Updated this week
• KellerJordan / modded-nanogpt
  NanoGPT (124M) in 3 minutes
  ☆2,501Updated this week
• patrick-kidger / jaxtyping
  Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/
  ☆1,385Updated this week
• arogozhnikov / einops
  Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
  ☆8,861Updated 2 months ago
• google-deepmind / optax
  Optax is a gradient processing and optimization library for JAX.
  ☆1,868Updated last week
• pytorch / tensordict
  TensorDict is a pytorch dedicated tensor container.
  ☆911Updated this week
• pytorch-labs / gpt-fast
  Simple and efficient pytorch-native transformer text generation in <1000 LOC of python.
  ☆5,930Updated 2 weeks ago
• dair-ai / Transformers-Recipe
  🧠 A study guide to learn about Transformers
  ☆1,576Updated last year
• lucidrains / x-transformers
  A concise but complete full-attention transformer with a set of promising experimental features from various papers
  ☆5,248Updated this week
• microsoft / mup
  maximal update parametrization (µP)
  ☆1,498Updated 9 months ago
• EleutherAI / cookbook
  Deep learning for dummies. All the practical details and useful utilities that go into working with real models.
  ☆786Updated last month