Alternatives and similar repositories for mm

Users that are interested in mm are comparing it to the libraries listed below

• srush / raspy
  An interactive exploration of Transformer programming.
  ☆265Updated last year
• jxbz / agd
  Automatic gradient descent
  ☆208Updated 2 years ago
• HenryNdubuaku / nanodl
  A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.
  ☆290Updated 10 months ago
• jax-ml / scaling-book
  Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs
  ☆424Updated this week
• dshah3 / GPU-Puzzles
  Solve puzzles. Learn CUDA.
  ☆64Updated last year
• srush / GPTWorld
  A puzzle to learn about prompting
  ☆131Updated 2 years ago
• jxmorris12 / gptzip
  Losslessly encode text natively with arithmetic coding and HuggingFace Transformers
  ☆76Updated 11 months ago
• vtabbott / Algebraic-NCD
  A package for defining deep learning models using categorical algebraic expressions.
  ☆61Updated 11 months ago
• idiap / sigma-gpt
  σ-GPT: A New Approach to Autoregressive Models
  ☆65Updated 11 months ago
• google-deepmind / nanodo
  ☆273Updated last year
• maxencefaldor / cax
  Cellular Automata Accelerated in JAX (Oral at ICLR 2025)
  ☆205Updated 2 months ago
• ulrichstern / cuda-convnet
  Alex Krizhevsky's original code from Google Code
  ☆194Updated 9 years ago
• RobertRiachi / nanoPALM
  ☆143Updated 2 years ago
• AI-Hypercomputer / maxdiffusion
  ☆230Updated this week
• mcleish7 / arithmetic
  Code to reproduce "Transformers Can Do Arithmetic with the Right Embeddings", McLeish et al (NeurIPS 2024)
  ☆190Updated last year
• sangmichaelxie / cs324_p2
  Project 2 (Building Large Language Models) for Stanford CS324: Understanding and Developing Large Language Models (Winter 2022)
  ☆105Updated 2 years ago
• character-ai / MuKoe
  ☆54Updated last year
• google / jaxonnxruntime
  A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.
  ☆115Updated 2 weeks ago
• lucidrains / flash-attention-jax
  Implementation of Flash Attention in Jax
  ☆213Updated last year
• google-deepmind / mishax
  ☆134Updated 3 months ago
• jax-ml / ml_dtypes
  A stand-alone implementation of several NumPy dtype extensions used in machine learning.
  ☆280Updated this week
• PolymathicAI / xVal
  Repository for code used in the xVal paper
  ☆136Updated last year
• NVIDIA / JAX-Toolbox
  JAX-Toolbox
  ☆321Updated this week
• KindXiaoming / grow-crystals
  Getting crystal-like representations with harmonic loss
  ☆191Updated 3 months ago
• vdesai2014 / inference-optimization-blog-post
  ☆88Updated last year
• allenai / fm-cheatsheet
  Website for hosting the Open Foundation Models Cheat Sheet.
  ☆267Updated 2 months ago
• mag- / gpu_benchmark
  Gpu benchmark
  ☆63Updated 5 months ago
• HazyResearch / train-tk
  train with kittens!
  ☆61Updated 8 months ago
• ml-gde / jflux
  JAX Implementation of Black Forest Labs' Flux.1 family of models
  ☆34Updated 8 months ago
• modula-systems / modula
  🧱 Modula software package
  ☆204Updated 3 months ago