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
• google-deepmind / spectral_ssm
  ☆31Updated 9 months ago
• bm371613 / conditional-flow-matching
  ☆14Updated 8 months ago
• nikhilvyas / SOAP
  ☆150Updated last month
• nZhangx / TrajectoryFlowMatching
  Code repository for Trajectory Flow Matching
  ☆50Updated 2 months ago
• itsdaniele / jeometric
  Graph neural networks in JAX.
  ☆67Updated 7 months ago
• lucidrains / grokfast-pytorch
  Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"
  ☆95Updated last month
• idiap / sigma-gpt
  σ-GPT: A New Approach to Autoregressive Models
  ☆61Updated 5 months ago
• TariqAHassan / S4Torch
  PyTorch implementation of Structured State Space for Sequence Modeling (S4), based on Annotated S4.
  ☆73Updated 10 months ago
• lucidrains / hyper-connections
  Attempt to make multiple residual streams from Bytedance's Hyper-Connections paper accessible to the public
  ☆66Updated last week
• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆83Updated 5 months ago
• epfml / DenseFormer
  ☆78Updated 9 months ago
• dvruette / barrel-rec-pytorch
  ☆53Updated last year
• google-deepmind / md4
  Code for https://arxiv.org/abs/2406.04329
  ☆56Updated last month
• helblazer811 / diffusion-visualizations
  Visualizations of the theory behind diffusion models.
  ☆77Updated 9 months ago
• phlippe / jax_trainer
  Lightning-like training API for JAX with Flax
  ☆38Updated last month
• radarFudan / mamba-minimal-jax
  ☆30Updated 2 months ago
• facebookresearch / lagrangian-ot
  Neural Optimal Transport with Lagrangian Costs
  ☆50Updated 6 months ago
• fal-ai-community / minDDPD
  ☆30Updated 3 weeks ago
• shikaiqiu / compute-better-spent
  ☆50Updated 3 months ago
• jmtomczak / vae_kan_example
  A simple example of VAEs with KANs
  ☆12Updated 8 months ago
• lucidrains / spline-based-transformer
  Implementation of the proposed Spline-Based Transformer from Disney Research
  ☆85Updated 2 months ago
• sidhu2690 / Deep-KAN
  This repository contains a better implementation of Kolmogorov-Arnold networks
  ☆59Updated 8 months ago
• jfpuget / ARC-AGI-Challenge-2024
  ☆46Updated 2 months ago
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆82Updated last year
• zacharyhorvitz / Fk-Diffusion-Steering
  A general framework for inference-time scaling and steering of diffusion models with arbitrary rewards.
  ☆71Updated 2 weeks ago
• PeaBrane / mamba-tiny
  Simple, minimal implementation of the Mamba SSM in one pytorch file. Using logcumsumexp (Heisen sequence).
  ☆106Updated 3 months ago
• lucidrains / scaling-vin-pytorch
  Exploration into the Scaling Value Iteration Networks paper, from Schmidhuber's group
  ☆36Updated 4 months ago
• kklemon / FlashPerceiver
  Fast and memory efficient PyTorch implementation of the Perceiver with FlashAttention.
  ☆24Updated 2 months ago
• andylolu2 / jax-diffusion
  Implementation of Denoising Diffusion Probabilistic Models (DDPM) in JAX and Flax.
  ☆17Updated last year