Add support for memoroids (linear recurrent models)#91

Open

smorad wants to merge 41 commits intoEdanToledo:mainfrom

smorad:memoroid

Commits on Jun 16, 2024

Convert ffm memoroid to flax, it seems to work
smorad
committed

Commits on Jun 21, 2024

Commits on Jun 22, 2024

Commits on Jun 23, 2024

feat: make edits to ffm to make batch dim after sequence dim and add back the training script that can use ffm
EdanToledo
committed
Merge branch 'main' into pr/smorad/91
EdanToledo
committed

Commits on Jun 26, 2024

fix: add required popjym wrapper
EdanToledo
committed

Commits on Jun 30, 2024

chore: move all current work into memoroids file to be contained
EdanToledo
committed

Commits on Jul 1, 2024

small hparam tweaks for memoroid
smorad
committed

Commits on Jul 3, 2024

Commits on Jul 4, 2024

Commits on Jul 7, 2024