Home

Awesome

PyTorch LMU

This repository contains PyTorch implementations of the following papers:

Performance on the psMNIST dataset is demonstrated in examples/.

Usage

torch, numpy, and scipy are the only requirements.
src/lmu.py contains the implementations of LMUCell, LMU and LMUFFT.

Examples:

Running on psMNIST

References