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
  ☆206Updated last year
• google-deepmind / nanodo
  ☆201Updated 6 months ago
• srush / raspy
  An interactive exploration of Transformer programming.
  ☆255Updated last year
• jxmorris12 / gptzip
  Losslessly encode text natively with arithmetic coding and HuggingFace Transformers
  ☆71Updated 5 months ago
• lucidrains / flash-attention-jax
  Implementation of Flash Attention in Jax
  ☆204Updated 10 months ago
• vtabbott / Algebraic-NCD
  A package for defining deep learning models using categorical algebraic expressions.
  ☆58Updated 5 months ago
• itsdaniele / jeometric
  Graph neural networks in JAX.
  ☆67Updated 6 months ago
• irregular-rhomboid / EAI-Math-Reading-Group
  Resources from the EleutherAI Math Reading Group
  ☆52Updated last month
• mcleish7 / arithmetic
  Code to reproduce "Transformers Can Do Arithmetic with the Right Embeddings", McLeish et al (NeurIPS 2024)
  ☆182Updated 7 months ago
• dshah3 / GPU-Puzzles
  Solve puzzles. Learn CUDA.
  ☆61Updated last year
• joey00072 / Tinytorch
  A really tiny autograd engine
  ☆89Updated 9 months ago
• HMUNACHI / nanodl
  A Jax-based library for designing and training transformer models from scratch.
  ☆280Updated 4 months ago
• RobertRiachi / nanoPALM
  ☆143Updated last year
• misko / human_descent
  ☆36Updated last month
• stanford-crfm / haliax
  Named Tensors for Legible Deep Learning in JAX
  ☆157Updated last week
• idiap / sigma-gpt
  σ-GPT: A New Approach to Autoregressive Models
  ☆61Updated 5 months ago
• srush / GPTWorld
  A puzzle to learn about prompting
  ☆123Updated last year
• young-geng / scalax
  A simple library for scaling up JAX programs
  ☆129Updated 2 months ago
• sangmichaelxie / cs324_p2
  Project 2 (Building Large Language Models) for Stanford CS324: Understanding and Developing Large Language Models (Winter 2022)
  ☆102Updated last year
• AllanYangZhou / midGPT
  Distributed pretraining of large language models (LLMs) on cloud TPU slices, with Jax and Equinox.
  ☆19Updated 3 months ago
• maxencefaldor / cax
  Cellular Automata Accelerated in JAX
  ☆77Updated last month
• samuela / torch2jax
  Run PyTorch in JAX. 🤝
  ☆214Updated last week
• cgarciae / einop
  ☆58Updated 2 years ago
• cloneofsimo / min-fsdp
  ☆75Updated 6 months ago
• modula-systems / modula
  🧱 Modula software package
  ☆132Updated this week
• cgarciae / nnx
  Neural Networks for JAX
  ☆83Updated 3 months ago
• lucidrains / PaLM-jax
  Implementation of the specific Transformer architecture from PaLM - Scaling Language Modeling with Pathways - in Jax (Equinox framework)
  ☆184Updated 2 years ago
• anas-rz / k3-node
  Multibackend Graph Neural Networks in Keras 3
  ☆25Updated 11 months ago
• ayaka14732 / llama-2-jax
  JAX implementation of the Llama 2 model
  ☆213Updated 11 months ago
• character-ai / MuKoe
  ☆52Updated 8 months ago