Lemon-cmd / energy-transformer-torchLinks
Official Implementation of Energy Transformer in PyTorch for Mask Image Reconstruction
☆32Updated last year
Alternatives and similar repositories for energy-transformer-torch
Users that are interested in energy-transformer-torch are comparing it to the libraries listed below
Sorting:
- Pytorch implementation of Simplified Structured State-Spaces for Sequence Modeling (S5)☆82Updated last year
- ☆314Updated last year
- Oscillatory State-Space Models☆113Updated 3 months ago
- Non official implementation of the Linear Recurrent Unit (LRU, Orvieto et al. 2023)☆61Updated 4 months ago
- FastKAN: Very Fast Implementation of Kolmogorov-Arnold Networks (KAN)☆467Updated last year
- This repo provides some examples (mostly solving PDE) to compare KAN and MLP☆19Updated last year
- Official implementation of Stochastic Taylor Derivative Estimator (STDE) NeurIPS2024☆128Updated last year
- Simple, minimal implementation of the Mamba SSM in one pytorch file. Using logcumsumexp (Heisen sequence).☆130Updated last year
- CodePDE: An Inference Framework for LLM-driven PDE Solver Generation☆65Updated 5 months ago
- This repository contains the official code for Energy Transformer---an efficient Energy-based Transformer variant for graph classificatio…☆25Updated 2 years ago
- Code to simulate energy-based analog systems and equilibrium propagation☆31Updated 9 months ago
- Kolmogorov-Arnold Networks (KAN) using Chebyshev polynomials instead of B-splines.☆402Updated last year
- [NeurIPS 2025 Spotlight] "Stochastic Process Learning via Operator Flow Matching"☆17Updated 2 months ago
- Liquid Structural State-Space Models☆383Updated 2 years ago
- [ICLR2025 Spotlight] Official implementation of Conflict-Free Inverse Gradients Method☆100Updated 2 months ago
- Minimal Mamba-2 implementation in PyTorch☆243Updated last year
- PyTorch implementation of Structured State Space for Sequence Modeling (S4), based on Annotated S4.☆88Updated last year
- ☆238Updated last year
- ICON for in-context operator learning☆60Updated 11 months ago
- Unofficial implementation of Linear Recurrent Units, by Deepmind, in Pytorch☆73Updated 9 months ago
- ☆140Updated last year
- tinybig for deep function learning☆60Updated 7 months ago
- Code for paper "Multiple Physics Pretraining for Physical Surrogate Models☆187Updated last year
- ☆255Updated 3 months ago
- An easy to use PyTorch implementation of the Kolmogorov Arnold Network and a few novel variations☆187Updated last year
- Pytorch implementation of the xLSTM model by Beck et al. (2024)☆181Updated last year
- ☆13Updated last year
- Reading list for research topics in state-space models☆341Updated 7 months ago
- Collection of papers on state-space models☆616Updated 2 months ago
- AL4PDE: A Benchmark for Active Learning for Neural PDE Solvers☆28Updated 8 months ago