lucidrains / kalman-filtering-attentionView external linksLinks
Implementation of the Kalman Filtering Attention proposed in "Kalman Filtering Attention for User Behavior Modeling in CTR Prediction"
☆59Oct 22, 2023Updated 2 years ago
Alternatives and similar repositories for kalman-filtering-attention
Users that are interested in kalman-filtering-attention are comparing it to the libraries listed below
Sorting:
- Yet another random morning idea to be quickly tried and architecture shared if it works; to allow the transformer to pause for any amount…☆53Oct 22, 2023Updated 2 years ago
- Implementation of CALM from the paper "LLM Augmented LLMs: Expanding Capabilities through Composition", out of Google Deepmind☆179Sep 12, 2024Updated last year
- Implementation of Agent Attention in Pytorch☆93Jul 10, 2024Updated last year
- Some personal experiments around routing tokens to different autoregressive attention, akin to mixture-of-experts☆123Oct 17, 2024Updated last year
- Explorations into the recently proposed Taylor Series Linear Attention☆100Aug 18, 2024Updated last year
- Implementation of the Llama architecture with RLHF + Q-learning☆170Feb 1, 2025Updated last year
- Implementation of Discrete Key / Value Bottleneck, in Pytorch☆88Jul 9, 2023Updated 2 years ago
- Implementation of Block Recurrent Transformer - Pytorch☆224Aug 20, 2024Updated last year
- Implementation of an Attention layer where each head can attend to more than just one token, using coordinate descent to pick topk☆47Jul 16, 2023Updated 2 years ago
- Implementation of a Light Recurrent Unit in Pytorch☆49Oct 6, 2024Updated last year
- Official implementation for the paper "EBMDOCK: NEURAL PROBABILISTIC PROTEIN-PROTEIN DOCKING VIA A DIFFERENTIABLE ENERGY-BASED MODEL" (IC…☆14Apr 24, 2024Updated last year
- Implementation of the transformer proposed in "Building Blocks for a Complex-Valued Transformer Architecture"☆88Oct 13, 2023Updated 2 years ago
- A paper list of world model☆29Apr 10, 2025Updated 10 months ago
- My explorations into editing the knowledge and memories of an attention network☆35Dec 8, 2022Updated 3 years ago
- Fine-Tuning Pre-trained Transformers into Decaying Fast Weights☆19Oct 9, 2022Updated 3 years ago
- Implementation of Flash Attention in Jax☆225Mar 1, 2024Updated last year
- A place to store reusable transformer components of my own creation or found on the interwebs☆73Updated this week
- ChemPile project☆19Jul 31, 2025Updated 6 months ago
- An interactive tool for analyzing, executing, and improving dynamic programming algorithms.☆19Jan 30, 2026Updated 2 weeks ago
- Simple Tensorflow implementation of "SRM : A Style-based Recalibration Module for Convolutional Neural Networks"☆18May 30, 2019Updated 6 years ago
- An attempt to merge ESBN with Transformers, to endow Transformers with the ability to emergently bind symbols☆16Aug 3, 2021Updated 4 years ago
- Memory-efficient optimum einsum using opt_einsum planning and PyTorch kernels.☆16Apr 24, 2023Updated 2 years ago
- MetaOpt: Towards efficient heuristic design with quantifiable and confident performance☆21Jan 20, 2026Updated 3 weeks ago
- Implementation of TableFormer, Robust Transformer Modeling for Table-Text Encoding, in Pytorch☆39Mar 29, 2022Updated 3 years ago
- Research Paper: "Graph Contrastive Learning as a Versatile Foundation for Advanced scRNA-seq Data Analysis"☆10Nov 20, 2024Updated last year
- CUDA implementation of autoregressive linear attention, with all the latest research findings☆46May 23, 2023Updated 2 years ago
- ☆17Nov 17, 2023Updated 2 years ago
- Multidimensional indexing for tensors☆137Jul 17, 2023Updated 2 years ago
- Implementation of a holodeck, written in Pytorch☆18Nov 1, 2023Updated 2 years ago
- Local Attention - Flax module for Jax☆22May 26, 2021Updated 4 years ago
- Pytorch Implementation of ShaDDR (SIGGRAPH Asia 2023 Conference)☆21Jan 24, 2025Updated last year
- Implementation of Invariant Point Attention, used for coordinate refinement in the structure module of Alphafold2, as a standalone Pytorc…☆171Nov 25, 2022Updated 3 years ago
- This is the official repo for Gradient Agreement Filtering (GAF).☆24Jan 27, 2025Updated last year
- ☆27Jul 28, 2025Updated 6 months ago
- Implementation of Token Shift GPT - An autoregressive model that solely relies on shifting the sequence space for mixing☆49Jan 27, 2022Updated 4 years ago
- Implementation of Recurrent Interface Network (RIN), for highly efficient generation of images and video without cascading networks, in P…☆207Feb 14, 2024Updated 2 years ago
- Code implementing "Efficient Parallelization of a Ubiquitious Sequential Computation" (Heinsen, 2023)☆98Dec 5, 2024Updated last year
- Implementation of Strassen attention, from Kozachinskiy et al. of National Center of AI in Chile☆41Jul 8, 2025Updated 7 months ago
- Posterior Control of Blackbox Generation☆23May 2, 2020Updated 5 years ago