Alternatives and similar repositories for mm

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

• jxbz / agd
  Automatic gradient descent
  ☆215Updated 2 years ago
• srush / raspy
  An interactive exploration of Transformer programming.
  ☆270Updated 2 years ago
• HenryNdubuaku / nanodl
  A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.
  ☆297Updated last year
• idiap / sigma-gpt
  σ-GPT: A New Approach to Autoregressive Models
  ☆70Updated last year
• joey00072 / Tinytorch
  A really tiny autograd engine
  ☆96Updated 6 months ago
• AI-Hypercomputer / maxdiffusion
  ☆288Updated this week
• RobertRiachi / nanoPALM
  ☆144Updated 2 years ago
• vdesai2014 / inference-optimization-blog-post
  ☆90Updated last year
• lucidrains / flash-attention-jax
  Implementation of Flash Attention in Jax
  ☆222Updated last year
• maxencefaldor / cax
  Cellular Automata Accelerated in JAX (Oral at ICLR 2025)
  ☆232Updated 2 weeks ago
• srush / GPTWorld
  A puzzle to learn about prompting
  ☆135Updated 2 years ago
• google-deepmind / nanodo
  ☆285Updated last year
• divyamakkar0 / JAXformer
  A zero-to-one guide on scaling modern transformers with n-dimensional parallelism.
  ☆105Updated 2 months ago
• google / jaxonnxruntime
  A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.
  ☆125Updated 2 months ago
• sumo43 / loopvlm
  run paligemma in real time
  ☆133Updated last year
• google-deepmind / mishax
  ☆144Updated 3 months ago
• jxmorris12 / gptzip
  Losslessly encode text natively with arithmetic coding and HuggingFace Transformers
  ☆76Updated last month
• abetlen / ggml-python
  Python bindings for ggml
  ☆146Updated last year
• profConradi / ALife
  Artificial Life simulations
  ☆54Updated 9 months ago
• dshah3 / GPU-Puzzles
  Solve puzzles. Learn CUDA.
  ☆64Updated 2 years ago
• imbue-ai / carbs
  Cost aware hyperparameter tuning algorithm
  ☆176Updated last year
• NVIDIA / JAX-Toolbox
  JAX-Toolbox
  ☆367Updated this week
• mcleish7 / arithmetic
  Code to reproduce "Transformers Can Do Arithmetic with the Right Embeddings", McLeish et al (NeurIPS 2024)
  ☆196Updated last year
• mag- / gpu_benchmark
  Gpu benchmark
  ☆73Updated 10 months 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
• itsdaniele / jeometric
  Graph neural networks in JAX.
  ☆68Updated last year
• MatX-inc / seqax
  seqax = sequence modeling + JAX
  ☆168Updated 4 months ago
• saurabhaloneai / Llama-3-From-Scratch-In-Pure-Jax
  This repository contain the simple llama3 implementation in pure jax.
  ☆70Updated 9 months ago
• irregular-rhomboid / EAI-Math-Reading-Group
  Resources from the EleutherAI Math Reading Group
  ☆54Updated 9 months ago
• NousResearch / StripedHyenaTrainer
  ☆62Updated 2 years ago