Alternatives and similar repositories for GPU-Puzzles

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

• cloneofsimo / min-fsdp
  ☆78Updated 10 months ago
• gpu-mode / profiling-cuda-in-torch
  ☆156Updated last year
• vdesai2014 / inference-optimization-blog-post
  ☆88Updated last year
• ethansmith2000 / fsdp_optimizers
  supporting pytorch FSDP for optimizers
  ☆79Updated 5 months ago
• linjames0 / Transformer-CUDA
  An implementation of the transformer architecture onto an Nvidia CUDA kernel
  ☆183Updated last year
• srush / GPTWorld
  A puzzle to learn about prompting
  ☆127Updated 2 years ago
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆78Updated last year
• MatX-inc / seqax
  seqax = sequence modeling + JAX
  ☆155Updated last month
• siboehm / ShallowSpeed
  Small scale distributed training of sequential deep learning models, built on Numpy and MPI.
  ☆133Updated last year
• srush / Transformer-Puzzles
  Puzzles for exploring transformers
  ☆347Updated 2 years ago
• srush / Autodiff-Puzzles
  ☆431Updated 7 months ago
• google-deepmind / nanodo
  ☆262Updated 10 months ago
• joey00072 / Tinytorch
  A really tiny autograd engine
  ☆95Updated this week
• Jaykef / Triton-nanoGPT
  Custom triton kernels for training Karpathy's nanoGPT.
  ☆19Updated 7 months ago
• dvruette / barrel-rec-pytorch
  ☆53Updated last year
• Deep-Learning-Profiling-Tools / triton-viz
  ☆210Updated this week
• cloneofsimo / ptx-tutorial-by-aislop
  PTX-Tutorial Written Purely By AIs (Deep Research of Openai and Claude 3.7)
  ☆67Updated 2 months ago
• MekkCyber / TritonAcademy
  A repository to unravel the language of GPUs, making their kernel conversations easy to understand
  ☆182Updated last week
• srush / raspy
  An interactive exploration of Transformer programming.
  ☆264Updated last year
• cloneofsimo / min-max-gpt
  Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training
  ☆127Updated last year
• HazyResearch / train-tk
  train with kittens!
  ☆57Updated 7 months ago
• gpu-mode / ring-attention
  ring-attention experiments
  ☆143Updated 7 months ago
• VatsaDev / NanoPoor
  NanoGPT-speedrunning for the poor T4 enjoyers
  ☆66Updated last month
• ayaka14732 / llama-2-jax
  JAX implementation of the Llama 2 model
  ☆217Updated last year
• gpu-mode / discord-cluster-manager
  Write a fast kernel and run it on Discord. See how you compare against the best!
  ☆44Updated this week
• tspeterkim / mixed-precision-from-scratch
  Mixed precision training from scratch with Tensors and CUDA
  ☆23Updated last year
• stas00 / ml-ways
  ML/DL Math and Method notes
  ☆61Updated last year
• huggingface / kernels
  Load compute kernels from the Hub
  ☆134Updated this week
• lessw2020 / triton_kernels_for_fun_and_profit
  Custom kernels in Triton language for accelerating LLMs
  ☆19Updated last year
• jax-ml / jax-llm-examples
  ☆112Updated last week