Alternatives and similar repositories for fourier-head:

Users that are interested in fourier-head are comparing it to the libraries listed below

• ruke1ire / RTF
  A State-Space Model with Rational Transfer Function Representation.
  ☆77Updated 8 months ago
• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆83Updated 6 months ago
• epfml / DenseFormer
  ☆78Updated 10 months ago
• google-deepmind / spectral_ssm
  ☆31Updated 10 months ago
• riverstone496 / awesome-second-order-optimization
  ☆25Updated last year
• dvruette / barrel-rec-pytorch
  ☆53Updated last year
• ml-jku / hopfield-boosting
  ☆29Updated 9 months ago
• lucidrains / hyper-connections
  Attempt to make multiple residual streams from Bytedance's Hyper-Connections paper accessible to the public
  ☆72Updated 2 weeks ago
• lucidrains / grokfast-pytorch
  Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"
  ☆95Updated last month
• TariqAHassan / S4Torch
  PyTorch implementation of Structured State Space for Sequence Modeling (S4), based on Annotated S4.
  ☆76Updated 11 months ago
• maxencefaldor / cax
  Cellular Automata Accelerated in JAX (Oral at ICLR 2025).
  ☆81Updated 2 months ago
• nikhilvyas / SOAP
  ☆157Updated 2 months ago
• PeaBrane / mamba-tiny
  Simple, minimal implementation of the Mamba SSM in one pytorch file. Using logcumsumexp (Heisen sequence).
  ☆108Updated 3 months ago
• lucidrains / spline-based-transformer
  Implementation of the proposed Spline-Based Transformer from Disney Research
  ☆85Updated 3 months ago
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆82Updated last year
• itsdaniele / jeometric
  Graph neural networks in JAX.
  ☆67Updated 7 months ago
• SHI-Labs / CompactNet
  ☆31Updated 9 months ago
• evanatyourservice / kron_torch
  An implementation of PSGD Kron second-order optimizer for PyTorch
  ☆80Updated this week
• bm371613 / conditional-flow-matching
  ☆14Updated 8 months ago
• idiap / sigma-gpt
  σ-GPT: A New Approach to Autoregressive Models
  ☆61Updated 6 months ago
• nZhangx / TrajectoryFlowMatching
  Code repository for Trajectory Flow Matching
  ☆51Updated 3 months ago
• alxndrTL / othello_mamba
  Evaluating the Mamba architecture on the Othello game
  ☆44Updated 9 months ago
• BlackHC / neural_net_checklist
  ☆149Updated 6 months ago
• helblazer811 / diffusion-visualizations
  Visualizations of the theory behind diffusion models.
  ☆77Updated 9 months ago
• NVlabs / GatedDeltaNet
  [ICLR 2025] Official PyTorch Implementation of Gated Delta Networks: Improving Mamba2 with Delta Rule
  ☆123Updated this week
• google-deepmind / md4
  Code for https://arxiv.org/abs/2406.04329
  ☆56Updated 2 months ago
• opallab / positional_attention
  Source code for the paper "Positional Attention: Out-of-Distribution Generalization and Expressivity for Neural Algorithmic Reasoning"
  ☆14Updated 2 weeks ago
• NVlabs / gbrl
  Gradient Boosting Reinforcement Learning (GBRL)
  ☆99Updated last week