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.
  ☆269Updated last year
• jxbz / agd
  Automatic gradient descent
  ☆215Updated 2 years ago
• jxmorris12 / gptzip
  Losslessly encode text natively with arithmetic coding and HuggingFace Transformers
  ☆76Updated last year
• vtabbott / Algebraic-NCD
  A package for defining deep learning models using categorical algebraic expressions.
  ☆61Updated last year
• sumo43 / loopvlm
  run paligemma in real time
  ☆133Updated last year
• HenryNdubuaku / nanodl
  A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.
  ☆295Updated last year
• mcleish7 / arithmetic
  Code to reproduce "Transformers Can Do Arithmetic with the Right Embeddings", McLeish et al (NeurIPS 2024)
  ☆193Updated last year
• divyamakkar0 / JAXformer
  A zero-to-one guide on scaling modern transformers with n-dimensional parallelism.
  ☆104Updated last month
• idiap / sigma-gpt
  σ-GPT: A New Approach to Autoregressive Models
  ☆68Updated last year
• vdesai2014 / inference-optimization-blog-post
  ☆89Updated last year
• RobertRiachi / nanoPALM
  ☆144Updated 2 years ago
• irregular-rhomboid / EAI-Math-Reading-Group
  Resources from the EleutherAI Math Reading Group
  ☆54Updated 8 months ago
• google-deepmind / nanodo
  ☆283Updated last year
• geohot / dumbrl
  Can RL solve simple problems?
  ☆54Updated last year
• lucidrains / flash-attention-jax
  Implementation of Flash Attention in Jax
  ☆219Updated last year
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆80Updated last year
• MatX-inc / seqax
  seqax = sequence modeling + JAX
  ☆168Updated 3 months ago
• maxencefaldor / cax
  Cellular Automata Accelerated in JAX (Oral at ICLR 2025)
  ☆227Updated this week
• google-deepmind / mishax
  ☆142Updated last month
• ulrichstern / cuda-convnet
  Alex Krizhevsky's original code from Google Code
  ☆199Updated 9 years ago
• joey00072 / Tinytorch
  A really tiny autograd engine
  ☆96Updated 5 months ago
• PolymathicAI / xVal
  Repository for code used in the xVal paper
  ☆144Updated last year
• HazyResearch / train-tk
  train with kittens!
  ☆63Updated last year
• dshah3 / GPU-Puzzles
  Solve puzzles. Learn CUDA.
  ☆64Updated last year
• character-ai / MuKoe
  ☆54Updated last year
• allenai / fm-cheatsheet
  Website for hosting the Open Foundation Models Cheat Sheet.
  ☆267Updated 5 months ago
• KhoomeiK / complexity-scaling
  gzip Predicts Data-dependent Scaling Laws
  ☆34Updated last year
• google-research / kauldron
  Modular, scalable library to train ML models
  ☆168Updated this week
• charlesfrye / cuda-substrings
  Because it's there.
  ☆16Updated last year
• google / jaxonnxruntime
  A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.
  ☆123Updated last month