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
- ☆313Updated last year
- 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)☆461Updated last year
- Code to simulate energy-based analog systems and equilibrium propagation☆31Updated 9 months ago
- Official implementation of Stochastic Taylor Derivative Estimator (STDE) NeurIPS2024☆125Updated last year
- Simple, minimal implementation of the Mamba SSM in one pytorch file. Using logcumsumexp (Heisen sequence).☆128Updated last year
- Unofficial implementation of Linear Recurrent Units, by Deepmind, in Pytorch☆73Updated 8 months ago
- This repo provides some examples (mostly solving PDE) to compare KAN and MLP☆19Updated last year
- Oscillatory State-Space Models☆112Updated 2 months ago
- Kolmogorov-Arnold Networks (KAN) using Chebyshev polynomials instead of B-splines.☆400Updated last year
- CodePDE: An Inference Framework for LLM-driven PDE Solver Generation☆63Updated 4 months ago
- Diffusion model derived evolutionary algorithm☆242Updated 6 months ago
- Parallelizing non-linear sequential models over the sequence length☆56Updated 6 months ago
- [NeurIPS 2025 Spotlight] "Stochastic Process Learning via Operator Flow Matching"☆15Updated 2 months ago
- Code for paper "Multiple Physics Pretraining for Physical Surrogate Models☆185Updated last year
- ICON for in-context operator learning☆60Updated 10 months ago
- This repository contains the official code for Energy Transformer---an efficient Energy-based Transformer variant for graph classificatio…☆25Updated last year
- Collection of papers on state-space models☆614Updated 2 months ago
- An easy to use PyTorch implementation of the Kolmogorov Arnold Network and a few novel variations☆186Updated last year
- ☆234Updated last year
- Implementation of "Gradients without backpropagation" paper (https://arxiv.org/abs/2202.08587) using functorch☆114Updated 2 years ago
- PyTorch implementation of Structured State Space for Sequence Modeling (S4), based on Annotated S4.☆87Updated last year
- ☆253Updated 2 months ago
- [ICLR2025 Spotlight] Official implementation of Conflict-Free Inverse Gradients Method☆98Updated last month
- Liquid Structural State-Space Models☆381Updated last year
- SR based on LLMs.☆124Updated 3 years ago
- Pytorch implementation of the xLSTM model by Beck et al. (2024)☆181Updated last year
- [ICLR'25] Artificial Kuramoto Oscillatory Neurons☆106Updated 2 months ago
- Code for "DPOT: Auto-Regressive Denoising Operator Transformer for Large-Scale PDE Pre-Training"☆67Updated last year