Example scripts to implement and train Mamba-2 on Trainium#36

Open

alexachille wants to merge 1 commit intoaws-neuron:mainfrom

alexachille:mamba2

Commits on Dec 2, 2024

Example scripts to implement and train a Mamba-2 architecture on Trainium.
alexachille
committed