/mamba-sandbox

Re-creating a simple implementation of Mamba using JAX and Pytorch.

Primary LanguageJupyter Notebook

mamba-sandbox