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
• RobertRiachi / nanoPALM
  ☆144Updated 2 years ago
• jxbz / agd
  Automatic gradient descent
  ☆210Updated 2 years ago
• joey00072 / Tinytorch
  A really tiny autograd engine
  ☆95Updated 3 months ago
• HenryNdubuaku / nanodl
  A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.
  ☆290Updated last year
• MatX-inc / seqax
  seqax = sequence modeling + JAX
  ☆166Updated last month
• sumo43 / loopvlm
  run paligemma in real time
  ☆131Updated last year
• ayaka14732 / llama-2-jax
  JAX implementation of the Llama 2 model
  ☆219Updated last year
• vdesai2014 / inference-optimization-blog-post
  ☆88Updated last year
• google-deepmind / nanodo
  ☆275Updated last year
• ulrichstern / cuda-convnet
  Alex Krizhevsky's original code from Google Code
  ☆196Updated 9 years ago
• rasbt / pytorch-memory-optim
  This code repository contains the code used for my "Optimizing Memory Usage for Training LLMs and Vision Transformers in PyTorch" blog po…
  ☆92Updated 2 years ago
• idiap / sigma-gpt
  σ-GPT: A New Approach to Autoregressive Models
  ☆67Updated last year
• maxencefaldor / cax
  Cellular Automata Accelerated in JAX (Oral at ICLR 2025)
  ☆211Updated 3 months ago
• jax-ml / ml_dtypes
  A stand-alone implementation of several NumPy dtype extensions used in machine learning.
  ☆292Updated 3 weeks ago
• NVIDIA / JAX-Toolbox
  JAX-Toolbox
  ☆332Updated this week
• google-research / kauldron
  Modular, scalable library to train ML models
  ☆156Updated this week
• sangmichaelxie / cs324_p2
  Project 2 (Building Large Language Models) for Stanford CS324: Understanding and Developing Large Language Models (Winter 2022)
  ☆105Updated 2 years ago
• vtabbott / Algebraic-NCD
  A package for defining deep learning models using categorical algebraic expressions.
  ☆61Updated last year
• dshah3 / GPU-Puzzles
  Solve puzzles. Learn CUDA.
  ☆64Updated last year
• facebookresearch / optimizers
  For optimization algorithm research and development.
  ☆530Updated last week
• srush / GPTWorld
  A puzzle to learn about prompting
  ☆132Updated 2 years ago
• google / jaxonnxruntime
  A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.
  ☆122Updated last month
• google-deepmind / mishax
  ☆141Updated 2 weeks ago
• allenai / fm-cheatsheet
  Website for hosting the Open Foundation Models Cheat Sheet.
  ☆267Updated 3 months ago
• mxbi / arckit
  Tools for working with the Abstraction & Reasoning Corpus
  ☆201Updated last week
• lucidrains / flash-attention-jax
  Implementation of Flash Attention in Jax
  ☆216Updated last year
• siboehm / ShallowSpeed
  Small scale distributed training of sequential deep learning models, built on Numpy and MPI.
  ☆136Updated last year
• jxmorris12 / gptzip
  Losslessly encode text natively with arithmetic coding and HuggingFace Transformers
  ☆76Updated last year
• kyegomez / swarms-pytorch
  Swarming algorithms like PSO, Ant Colony, Sakana, and more in PyTorch 😊
  ☆129Updated 3 weeks ago