Alternatives and similar repositories for mm

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

• google-deepmind / nanodo
  ☆270Updated 11 months ago
• dshah3 / GPU-Puzzles
  Solve puzzles. Learn CUDA.
  ☆64Updated last year
• lucidrains / flash-attention-jax
  Implementation of Flash Attention in Jax
  ☆213Updated last year
• srush / raspy
  An interactive exploration of Transformer programming.
  ☆264Updated last year
• srush / Transformer-Puzzles
  Puzzles for exploring transformers
  ☆350Updated 2 years ago
• srush / GPTWorld
  A puzzle to learn about prompting
  ☆128Updated 2 years ago
• jxbz / agd
  Automatic gradient descent
  ☆208Updated last year
• HMUNACHI / nanodl
  A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.
  ☆288Updated 9 months ago
• vtabbott / Algebraic-NCD
  A package for defining deep learning models using categorical algebraic expressions.
  ☆61Updated 10 months ago
• jax-ml / scaling-book
  Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs
  ☆399Updated 2 weeks ago
• cgarciae / einop
  ☆60Updated 3 years ago
• itsdaniele / jeometric
  Graph neural networks in JAX.
  ☆67Updated last year
• ulrichstern / cuda-convnet
  Alex Krizhevsky's original code from Google Code
  ☆192Updated 9 years ago
• idiap / sigma-gpt
  σ-GPT: A New Approach to Autoregressive Models
  ☆65Updated 10 months ago
• jxmorris12 / gptzip
  Losslessly encode text natively with arithmetic coding and HuggingFace Transformers
  ☆76Updated 10 months ago
• irregular-rhomboid / EAI-Math-Reading-Group
  Resources from the EleutherAI Math Reading Group
  ☆53Updated 3 months ago
• MatX-inc / seqax
  seqax = sequence modeling + JAX
  ☆159Updated last week
• vdesai2014 / inference-optimization-blog-post
  ☆88Updated last year
• stanford-crfm / haliax
  Named Tensors for Legible Deep Learning in JAX
  ☆181Updated this week
• joey00072 / Tinytorch
  A really tiny autograd engine
  ☆94Updated 3 weeks ago
• RobertRiachi / nanoPALM
  ☆143Updated 2 years ago
• kvfrans / jax-diffusion-transformer
  Implementation of Diffusion Transformer (DiT) in JAX
  ☆278Updated last year
• llm-random / llm-random
  ☆190Updated this week
• jax-ml / jax-triton
  jax-triton contains integrations between JAX and OpenAI Triton
  ☆400Updated 3 weeks ago
• cloneofsimo / min-fsdp
  ☆78Updated 11 months ago
• HazyResearch / zoology
  Understand and test language model architectures on synthetic tasks.
  ☆217Updated 2 weeks ago
• NVIDIA / JAX-Toolbox
  JAX-Toolbox
  ☆311Updated this week
• awf / functional-transformer
  A pure-functional implementation of a machine learning transformer model in Python/JAX
  ☆179Updated last month
• ayaka14732 / llama-2-jax
  JAX implementation of the Llama 2 model
  ☆218Updated last year
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆79Updated last year