tk-rusch / LEMLinks
Official code for Long Expressive Memory (ICLR 2022, Spotlight)
☆71Updated 3 years ago
Alternatives and similar repositories for LEM
Users that are interested in LEM are comparing it to the libraries listed below
Sorting:
- Code repository of the paper "CKConv: Continuous Kernel Convolution For Sequential Data" published at ICLR 2022. https://arxiv.org/abs/21…☆123Updated 2 years ago
- Sequence Modeling with Multiresolution Convolutional Memory (ICML 2023)☆127Updated 2 years ago
- Sequence Modeling with Structured State Spaces☆66Updated 3 years ago
- Code repository of the paper "Wavelet Networks: Scale-Translation Equivariant Learning From Raw Time-Series, TMLR" https://arxiv.org/abs…☆83Updated last year
- ☆31Updated 4 years ago
- Layerwise Batch Entropy Regularization☆23Updated 3 years ago
- Continuous Augmented Positional Embeddings (CAPE) implementation for PyTorch☆42Updated 2 years ago
- AdaCat☆49Updated 3 years ago
- An implementation of soft-DTW divergences.☆135Updated 4 years ago
- Implementation of Gated State Spaces, from the paper "Long Range Language Modeling via Gated State Spaces", in Pytorch☆101Updated 2 years ago
- ☆164Updated 2 years ago
- Relative Positional Encoding for Transformers with Linear Complexity☆65Updated 3 years ago
- A home for audio ML in JAX. Has common features, learnable frontends, pretrained supervised and self-supervised models.☆69Updated 3 years ago
- Code for ICLR 2021 Paper, "Anytime Sampling for Autoregressive Models via Ordered Autoencoding"☆26Updated 2 years ago
- Implementation of a Transformer that Ponders, using the scheme from the PonderNet paper☆81Updated 3 years ago
- Code for the paper PermuteFormer☆42Updated 4 years ago
- Implementation of the Kalman Filtering Attention proposed in "Kalman Filtering Attention for User Behavior Modeling in CTR Prediction"☆59Updated 2 years ago
- Implementations of various linear RNN layers using pytorch and triton☆54Updated 2 years ago
- The official implementation of the ChordMixer architecture.☆61Updated 2 years ago
- Code for: "Neural Rough Differential Equations for Long Time Series", (ICML 2021)☆118Updated 4 years ago
- repo for paper: Adaptive Checkpoint Adjoint (ACA) method for gradient estimation in neural ODE☆56Updated 4 years ago
- Easy-to-use AdaHessian optimizer (PyTorch)☆79Updated 4 years ago
- Implementation of GateLoop Transformer in Pytorch and Jax☆90Updated last year
- Implementation of Insertion-deletion Denoising Diffusion Probabilistic Models☆30Updated 3 years ago
- Local Attention - Flax module for Jax☆22Updated 4 years ago
- The accompanying code for "Simplifying and Understanding State Space Models with Diagonal Linear RNNs" (Ankit Gupta, Harsh Mehta, Jonatha…☆23Updated 2 years ago
- Official code for UnICORNN (ICML 2021)☆28Updated 4 years ago
- Implementation of Flow++ in PyTorch☆40Updated 6 years ago
- ☆33Updated 2 years ago
- Implementation of Mega, the Single-head Attention with Multi-headed EMA architecture that currently holds SOTA on Long Range Arena☆206Updated 2 years ago