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.
  ☆267Updated last year
• google-deepmind / nanodo
  ☆275Updated last year
• dshah3 / GPU-Puzzles
  Solve puzzles. Learn CUDA.
  ☆64Updated last year
• jxbz / agd
  Automatic gradient descent
  ☆208Updated 2 years ago
• vtabbott / Algebraic-NCD
  A package for defining deep learning models using categorical algebraic expressions.
  ☆61Updated last year
• HenryNdubuaku / nanodl
  A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.
  ☆290Updated 11 months ago
• AI-Hypercomputer / maxdiffusion
  ☆243Updated this week
• jxmorris12 / gptzip
  Losslessly encode text natively with arithmetic coding and HuggingFace Transformers
  ☆76Updated last year
• NVIDIA / JAX-Toolbox
  JAX-Toolbox
  ☆329Updated this week
• RobertRiachi / nanoPALM
  ☆143Updated 2 years ago
• joey00072 / Tinytorch
  A really tiny autograd engine
  ☆95Updated 2 months ago
• mcleish7 / arithmetic
  Code to reproduce "Transformers Can Do Arithmetic with the Right Embeddings", McLeish et al (NeurIPS 2024)
  ☆190Updated last year
• srush / GPTWorld
  A puzzle to learn about prompting
  ☆132Updated 2 years ago
• idiap / sigma-gpt
  σ-GPT: A New Approach to Autoregressive Models
  ☆67Updated 11 months ago
• jax-ml / scaling-book
  Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs
  ☆466Updated this week
• itsdaniele / jeometric
  Graph neural networks in JAX.
  ☆67Updated last year
• google / jaxonnxruntime
  A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.
  ☆119Updated last week
• MatX-inc / seqax
  seqax = sequence modeling + JAX
  ☆165Updated 2 weeks ago
• allenai / fm-cheatsheet
  Website for hosting the Open Foundation Models Cheat Sheet.
  ☆267Updated 3 months ago
• ml-gde / jflux
  JAX Implementation of Black Forest Labs' Flux.1 family of models
  ☆35Updated 9 months ago
• nebius / kvax
  A FlashAttention implementation for JAX with support for efficient document mask computation and context parallelism.
  ☆135Updated 4 months ago
• ayaka14732 / llama-2-jax
  JAX implementation of the Llama 2 model
  ☆219Updated last year
• google-research / kauldron
  Modular, scalable library to train ML models
  ☆143Updated this week
• google-deepmind / mishax
  ☆136Updated 4 months ago
• jax-ml / ml_dtypes
  A stand-alone implementation of several NumPy dtype extensions used in machine learning.
  ☆286Updated last week
• stanford-crfm / haliax
  Named Tensors for Legible Deep Learning in JAX
  ☆199Updated last week
• sumo43 / loopvlm
  run paligemma in real time
  ☆131Updated last year
• irregular-rhomboid / EAI-Math-Reading-Group
  Resources from the EleutherAI Math Reading Group
  ☆53Updated 5 months ago
• hundredblocks / large-model-parallelism
  Functional local implementations of main model parallelism approaches
  ☆96Updated 2 years ago
• yixiaoer / mistral-v0.2-jax
  JAX implementation of the Mistral 7b v0.2 model
  ☆35Updated last year