Alternatives and similar repositories for Metal-Puzzles

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

• obadakhalili / tinygrad-tensor-puzzles
  Solve puzzles to improve your tinygrad skills!
  ☆140Updated 4 months ago
• mesozoic-egg / tinygrad-notes
  Tutorials on tinygrad
  ☆396Updated last month
• mlecauchois / micrograd-cuda
  ☆249Updated last year
• smolorg / smolgrad
  small auto-grad engine inspired from Karpathy's micrograd and PyTorch
  ☆274Updated 8 months ago
• pranavjad / mlx-gpt2
  gpt-2 from scratch in mlx
  ☆394Updated last year
• tugot17 / pmpp
  Complete solutions to the Programming Massively Parallel Processors Edition 4
  ☆444Updated last month
• philipturner / metal-flash-attention
  FlashAttention (Metal Port)
  ☆512Updated 10 months ago
• Quentin-Anthony / torch-profiling-tutorial
  ☆447Updated 2 weeks ago
• arpitingle / gpu-alpha
  High Quality Resources on GPU Programming/Architecture
  ☆588Updated last year
• ml-explore / mlx-data
  Efficient framework-agnostic data loading
  ☆430Updated 2 months ago
• PaulPauls / llama3_interpretability_sae
  A complete end-to-end pipeline for LLM interpretability with sparse autoencoders (SAEs) using Llama 3.2, written in pure PyTorch and full…
  ☆620Updated 4 months ago
• EurekaLabsAI / tensor
  The Tensor (or Array)
  ☆441Updated 11 months ago
• clu0 / unet.cu
  UNet diffusion model in pure CUDA
  ☆613Updated last year
• Laz4rz / GPT-2
  Following master Karpathy with GPT-2 implementation and training, writing lots of comments cause I have memory of a goldfish
  ☆172Updated last year
• thomasahle / tensorgrad
  Machine Learning with Symbolic Tensors
  ☆324Updated 2 months ago
• MarioSieg / magnetron
  (WIP) A small but powerful, homemade PyTorch from scratch.
  ☆554Updated this week
• yousef-rafat / miniDiffusion
  A reimplementation of Stable Diffusion 3.5 in pure PyTorch
  ☆652Updated last month
• omkaark / simple-federated-learning
  ☆96Updated last year
• riccardomusmeci / mlx-llm
  Large Language Models (LLMs) applications and tools running on Apple Silicon in real-time with Apple MLX.
  ☆451Updated 6 months ago
• felafax / felafax
  Felafax is building AI infra for non-NVIDIA GPUs
  ☆566Updated 6 months ago
• google-deepmind / penzai
  A JAX research toolkit for building, editing, and visualizing neural networks.
  ☆1,806Updated last month
• geohotstan / tinycorp-meetings
  ☆87Updated last week
• google-deepmind / treescope
  An interactive HTML pretty-printer for machine learning research in IPython notebooks.
  ☆426Updated 3 months ago
• andrewn6 / fromthetransistor
  From the Transistor to the Web Browser, a rough outline for a 12 week course
  ☆232Updated last year
• wentasah / mmul-anim
  Visualization of cache-optimized matrix multiplication
  ☆153Updated 4 months ago
• xjdr-alt / simple_transformer
  Simple Transformer in Jax
  ☆138Updated last year
• haraschax / nograd
  Gradient descent is cool and all, but what if we could delete it?
  ☆104Updated last week
• srush / Autodiff-Puzzles
  ☆443Updated 9 months ago
• google-ai-edge / model-explorer
  A modern model graph visualizer and debugger
  ☆1,293Updated this week
• srush / Transformer-Puzzles
  Puzzles for exploring transformers
  ☆356Updated 2 years ago