tk-rusch / LEM
Official code for Long Expressive Memory (ICLR 2022, Spotlight)
☆69Updated 3 years ago
Alternatives and similar repositories for LEM:
Users that are interested in LEM are comparing it to the libraries listed below
- Sequence Modeling with Structured State Spaces☆63Updated 2 years ago
- Code repository of the paper "CKConv: Continuous Kernel Convolution For Sequential Data" published at ICLR 2022. https://arxiv.org/abs/21…☆119Updated 2 years ago
- AdaCat☆49Updated 2 years ago
- Sequence Modeling with Multiresolution Convolutional Memory (ICML 2023)☆122Updated last year
- Continuous Augmented Positional Embeddings (CAPE) implementation for PyTorch☆39Updated 2 years ago
- ☆31Updated 3 years ago
- Layerwise Batch Entropy Regularization☆22Updated 2 years ago
- Supplementary code for the paper "Meta-Solver for Neural Ordinary Differential Equations" https://arxiv.org/abs/2103.08561☆24Updated 3 years ago
- repo for paper: Adaptive Checkpoint Adjoint (ACA) method for gradient estimation in neural ODE☆54Updated 4 years ago
- Code repository of the paper "Wavelet Networks: Scale-Translation Equivariant Learning From Raw Time-Series, TMLR" https://arxiv.org/abs…☆81Updated last year
- [ICML 2024] SIRFShampoo: Structured inverse- and root-free Shampoo in PyTorch (https://arxiv.org/abs/2402.03496)☆14Updated 4 months ago
- An implementation of soft-DTW divergences.☆134Updated 3 years ago
- Official code for Coupled Oscillatory RNN (ICLR 2021, Oral)☆43Updated 3 years ago
- Non official implementation of the Linear Recurrent Unit (LRU, Orvieto et al. 2023)☆52Updated 4 months ago
- Code for: "Neural Rough Differential Equations for Long Time Series", (ICML 2021)☆116Updated 3 years ago
- Codes accompanying the paper "LaProp: a Better Way to Combine Momentum with Adaptive Gradient"☆27Updated 4 years ago
- ☆33Updated 2 years ago
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆83Updated last year
- Inspired by "Neural Networks Fail to Learn Periodic Functions and How to Fix It"☆63Updated 10 months ago
- Official code for UnICORNN (ICML 2021)☆27Updated 3 years ago
- [TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivative…☆17Updated last year
- Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)☆86Updated 2 years ago
- Implementation of Gated State Spaces, from the paper "Long Range Language Modeling via Gated State Spaces", in Pytorch☆99Updated 2 years ago
- A PyTorch implementation of Legendre Memory Units (LMUs) and its FFT variant☆41Updated 3 years ago
- ☆38Updated last year
- code for "Semi-Discrete Normalizing Flows through Differentiable Tessellation"☆26Updated 2 years ago
- Code for ICLR 2021 Paper, "Anytime Sampling for Autoregressive Models via Ordered Autoencoding"☆26Updated last year
- Code for the paper PermuteFormer☆42Updated 3 years ago
- ☆36Updated last year
- Implementations of various linear RNN layers using pytorch and triton☆50Updated last year