Alternatives and similar repositories for mm:

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

• joey00072 / Tinytorch
  A really tiny autograd engine
  ☆91Updated last year
• srush / raspy
  An interactive exploration of Transformer programming.
  ☆262Updated last year
• HMUNACHI / NanoDL
  A Jax-based library for designing and training small transformers.
  ☆286Updated 7 months ago
• idiap / sigma-gpt
  σ-GPT: A New Approach to Autoregressive Models
  ☆62Updated 8 months ago
• vtabbott / Algebraic-NCD
  A package for defining deep learning models using categorical algebraic expressions.
  ☆60Updated 8 months ago
• dshah3 / GPU-Puzzles
  Solve puzzles. Learn CUDA.
  ☆63Updated last year
• jxmorris12 / gptzip
  Losslessly encode text natively with arithmetic coding and HuggingFace Transformers
  ☆73Updated 8 months ago
• stanford-crfm / haliax
  Named Tensors for Legible Deep Learning in JAX
  ☆172Updated this week
• google-deepmind / nanodo
  ☆215Updated 9 months ago
• jax-ml / jax-triton
  jax-triton contains integrations between JAX and OpenAI Triton
  ☆390Updated last week
• modula-systems / modula
  🧱 Modula software package
  ☆188Updated 3 weeks ago
• vdesai2014 / inference-optimization-blog-post
  ☆87Updated last year
• jax-ml / scaling-book
  Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs
  ☆242Updated this week
• jxbz / agd
  Automatic gradient descent
  ☆207Updated last year
• mag- / gpu_benchmark
  Gpu benchmark
  ☆59Updated 2 months ago
• itsdaniele / jeometric
  Graph neural networks in JAX.
  ☆67Updated 10 months ago
• google-deepmind / mishax
  ☆128Updated 2 weeks ago
• cgarciae / einop
  ☆60Updated 3 years ago
• ayaka14732 / llama-2-jax
  JAX implementation of the Llama 2 model
  ☆218Updated last year
• kroggen / mamba.c
  Inference of Mamba models in pure C
  ☆187Updated last year
• lucidrains / flash-attention-jax
  Implementation of Flash Attention in Jax
  ☆206Updated last year
• irregular-rhomboid / EAI-Math-Reading-Group
  Resources from the EleutherAI Math Reading Group
  ☆53Updated last month
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆78Updated last year
• kvfrans / jax-diffusion-transformer
  Implementation of Diffusion Transformer (DiT) in JAX
  ☆270Updated 10 months ago
• AI-Hypercomputer / maxdiffusion
  ☆201Updated this week
• gpu-mode / discord-cluster-manager
  Write a fast kernel and run it on Discord. See how you compare against the best!
  ☆38Updated this week
• NVIDIA / JAX-Toolbox
  JAX-Toolbox
  ☆298Updated this week
• cloneofsimo / min-fsdp
  ☆77Updated 9 months ago
• jax-ml / jax-llm-examples
  ☆99Updated this week
• srush / Transformer-Puzzles
  Puzzles for exploring transformers
  ☆343Updated last year