Code

Add Mamab2 and its Jamba variant#839

Code

Open

berlino wants to merge 4 commits intoapple:mainapple/axlearn:mainfrom berlino:mamba2berlino/axlearn:mamba2Copy head branch name to clipboard

+2,821-5

Commits

Commits on Nov 14, 2024

• add mamab2

  

  bailin_wang

  committed
  fd6b372

Commits on Dec 4, 2024

• Merge branch 'main' into mamba2

  berlino

  authored
  1469a35

• merge

  

  bailin_wang

  committed
  0bb4ac3

Commits on Dec 12, 2024

• unify init and prefill

  

  bailin_wang

  committed
  678e392