Code

Add Mamab2 and its Jamba variant#839

Code

Merged

berlino merged 5 commits intoapple:mainapple/axlearn:mainfrom berlino:mamba2berlino/axlearn:mamba2Copy head branch name to clipboardJan 17, 2025

+2,827-6

Commits

Commits on Nov 14, 2024

• add mamab2

  

  bailin_wang

  committed
  fd6b372

  Copy full SHA for fd6b372

Commits on Dec 4, 2024

• Merge branch 'main' into mamba2

  berlino

  authored
  1469a35

  Copy full SHA for 1469a35

• merge

  

  bailin_wang

  committed
  0bb4ac3

  Copy full SHA for 0bb4ac3

Commits on Dec 12, 2024

• unify init and prefill

  

  bailin_wang

  committed
  678e392

  Copy full SHA for 678e392

Commits on Jan 17, 2025

• adapt final changes

  

  bailin_wang

  committed
  50870c5

  Copy full SHA for 50870c5