memax

memax: A library for memory-augmented neural networks.

We provide backends for both Equinox and Flax.

We utilize a Generalized Recurrent Algebraic Structure (GRAS) framework to define and implement virtually any memory model we come across using a single interface. Take a look at memax.equinox.gras or memax.linen.gras for more details.

View Source

1"""
2memax: A library for memory-augmented neural networks.
3
4We provide backends for both Equinox and Flax.
5
6We utilize a Generalized Recurrent Algebraic Structure (GRAS) framework
7to define and implement virtually any memory model we come across using a single interface.
8Take a look at `memax.equinox.gras` or `memax.linen.gras` for more details.
9"""