Alternatives and similar repositories for attention-approximates-sdm

Users that are interested in attention-approximates-sdm are comparing it to the libraries listed below

• spencerkent / resonator-networks
  An implementation of Resonator Networks
  ☆21Updated 5 years ago
• bhoov / energy-transformer-jax
  The Energy Transformer block, in JAX
  ☆63Updated 2 years ago
• thebuckleylab / jpc
  Flexible Inference for Predictive Coding Networks in JAX.
  ☆64Updated this week
• bhoov / hamux
  Hierarchical Associative Memory User Experience
  ☆105Updated last month
• RobertRosenbaum / Torch2PC
  Software for using predictive coding algorithms to train PyTorch models.
  ☆26Updated last year
• dylandoblar / noether-networks
  Meta-learning inductive biases in the form of useful conserved quantities.
  ☆39Updated 3 years ago
• romanpogodin / towards-bio-plausible-conv
  ☆14Updated 4 years ago
• julestalloen / pytorch-hebbian
  A lightweight and flexible framework for Hebbian learning in PyTorch.
  ☆95Updated last year
• akandykeller / NeuralWaveMachines
  Official Implementation of the ICML 2023 paper: "Neural Wave Machines: Learning Spatiotemporally Structured Representations with Locally …
  ☆77Updated 2 years ago
• BerenMillidge / PredictiveCodingBackprop
  Code for the paper "Predictive Coding Approximates Backprop along Arbitrary Computation Graphs"
  ☆165Updated 5 years ago
• google-deepmind / neural_testbed
  ☆192Updated 6 months ago
• liukidar / pcx
  Predictive Coding JAX-based library
  ☆86Updated 8 months ago
• conorheins / bayesian-mechanics-sdes
  ☆14Updated 3 years ago
• geometric-intelligence / neurometry
  Quantify geometric intelligence in natural and artificial brains.
  ☆51Updated 6 months ago
• 8bitmp3 / JAX-Flax-Tutorial-Image-Classification-with-Linen
  How to use the Flax Linen API to build a convolutional neural network model and train it for image classification (using TensorFlow Datas…
  ☆24Updated 2 years ago
• aks2203 / easy-to-hard
  Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"
  ☆60Updated 3 years ago
• leonard-gleyzer / connex
  Fine-grained, dynamic control of neural network topology in JAX.
  ☆21Updated 2 years ago
• FutureComputing4AI / Learning-with-Holographic-Reduced-Representations
  ☆35Updated 4 years ago
• ANazaret / unbounded-depth-neural-networks
  Implementation of the unbounded depth neural network from the paper Variational Inference for Infinitely Deep Neural Networks
  ☆17Updated 3 years ago
• google-deepmind / PGMax
  Loopy belief propagation for factor graphs on discrete variables in JAX
  ☆165Updated last year
• mcbal / afem
  Implementation of approximate free-energy minimization in PyTorch
  ☆20Updated 4 years ago
• ExpectationMax / sklearn-jax-kernels
  Composable kernels for scikit-learn implemented in JAX.
  ☆45Updated 5 years ago
• hadivafaii / IterativeVAE
  The Official PyTorch Implementation of "Brain-like Variational Inference" (NeurIPS 2025 Paper)
  ☆64Updated last month
• ejmichaud / grokking-squared
  ☆27Updated 2 years ago
• GallagherCommaJack / modulax
  ☆17Updated last year
• NACLab / ngc-learn
  NGC-Learn: Neurobiological Systems Simulation and NeuroAI Design in Python
  ☆170Updated last week
• MahmudulAlam / Holographic-Reduced-Representations
  Holographic Reduced Representations
  ☆28Updated 3 weeks ago
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆92Updated last year
• google-deepmind / tf2jax
  ☆118Updated 2 weeks ago
• mcbal / deep-implicit-attention
  Implementation of deep implicit attention in PyTorch
  ☆65Updated 4 years ago