hrshtv / pytorch-lmu
A PyTorch implementation of Legendre Memory Units (LMUs) and its FFT variant
☆41Updated 3 years ago
Alternatives and similar repositories for pytorch-lmu:
Users that are interested in pytorch-lmu are comparing it to the libraries listed below
- Unofficial implementation of Linear Recurrent Units, by Deepmind, in Pytorch☆64Updated last year
- Non official implementation of the Linear Recurrent Unit (LRU, Orvieto et al. 2023)☆51Updated 3 months ago
- Official code for Coupled Oscillatory RNN (ICLR 2021, Oral)☆43Updated 3 years ago
- Pytorch implementation of Simplified Structured State-Spaces for Sequence Modeling (S5)☆73Updated 9 months ago
- ☆14Updated 3 years ago
- Official code for Long Expressive Memory (ICLR 2022, Spotlight)☆69Updated 2 years ago
- Bayesian learning and inference for state space models (SSMs) using Google Research's JAX as a backend☆58Updated 7 months ago
- repo for code for paper on general theory associative memory models☆18Updated 2 years ago
- Implementation of the Legendre Memory Unit in PyTorch☆22Updated 5 years ago
- A lightweight and flexible framework for Hebbian learning in PyTorch.☆81Updated 11 months ago
- Official code for UnICORNN (ICML 2021)☆27Updated 3 years ago
- ☆37Updated last year
- Code for the paper "Predictive Coding Approximates Backprop along Arbitrary Computation Graphs"☆151Updated 4 years ago
- ☆13Updated 2 years ago
- PyTorch implementation of Structured State Space for Sequence Modeling (S4), based on Annotated S4.☆76Updated 11 months ago
- ☆35Updated 2 years ago
- Parallelizing non-linear sequential models over the sequence length☆50Updated 3 weeks ago
- Exca - Execution and caching tool for python☆74Updated this week
- ☆9Updated this week
- ☆28Updated 7 months ago
- Code for Hidden Markov Nonlinear ICA☆24Updated 3 years ago
- Code for our paper: Online Variational Filtering and Parameter Learning☆18Updated 3 years ago
- Code for the paper 'Disentangling Identifiable Features from Noisy Data with Structured Nonlinear ICA' @ Neurips'21☆18Updated this week
- Automatic Hebbian learning in multi-layer convolutional networks with PyTorch, by expressing Hebbian plasticity rules as gradients☆37Updated last year
- A collection of meta-learning algorithms in Jax☆23Updated 2 years ago
- The accompanying code for "Simplifying and Understanding State Space Models with Diagonal Linear RNNs" (Ankit Gupta, Harsh Mehta, Jonatha…☆19Updated 2 years ago
- Dreamer on JAX☆15Updated 3 years ago
- Fully documented Pytorch implementation of the Equilibrium Propagation algorithm.☆32Updated 5 years ago
- ☆9Updated 3 years ago
- Sequence Modeling with Multiresolution Convolutional Memory (ICML 2023)☆122Updated last year