README.md
pyproject.toml
setup.py
memax/__init__.py
memax/mtypes.py
memax/utils.py
memax.egg-info/PKG-INFO
memax.egg-info/SOURCES.txt
memax.egg-info/dependency_links.txt
memax.egg-info/requires.txt
memax.egg-info/top_level.txt
memax/equinox/__init__.py
memax/equinox/gras.py
memax/equinox/groups.py
memax/equinox/scans.py
memax/equinox/train_utils.py
memax/equinox/models/__init__.py
memax/equinox/models/residual.py
memax/equinox/semigroups/__init__.py
memax/equinox/semigroups/attn.py
memax/equinox/semigroups/delta.py
memax/equinox/semigroups/deltap.py
memax/equinox/semigroups/dlse.py
memax/equinox/semigroups/fart.py
memax/equinox/semigroups/ffm.py
memax/equinox/semigroups/fwp.py
memax/equinox/semigroups/gdn.py
memax/equinox/semigroups/lrnn.py
memax/equinox/semigroups/lru.py
memax/equinox/semigroups/mlp.py
memax/equinox/semigroups/nmax.py
memax/equinox/semigroups/s6.py
memax/equinox/semigroups/spherical.py
memax/equinox/semigroups/stack.py
memax/equinox/set_actions/__init__.py
memax/equinox/set_actions/elman.py
memax/equinox/set_actions/gru.py
memax/equinox/set_actions/lstm.py
memax/equinox/set_actions/mgu.py
memax/equinox/set_actions/spherical.py
memax/linen/__init__.py
memax/linen/gras.py
memax/linen/groups.py
memax/linen/scans.py
memax/linen/train_utils.py
memax/linen/models/__init__.py
memax/linen/models/residual.py
memax/linen/semigroups/__init__.py
memax/linen/semigroups/fart.py
memax/linen/semigroups/lru.py
memax/linen/semigroups/s6.py
memax/linen/set_actions/__init__.py
memax/linen/set_actions/gru.py
tests/test_associative_equinox.py
tests/test_associative_linen.py
tests/test_continuous_localization.py
tests/test_initial_input_equinox.py
tests/test_initial_input_linen.py
tests/test_readme.py
tests/test_reset_equinox.py
tests/test_reset_linen.py
tests/test_stack_equinox.py