Alternatives and similar repositories for modern-srwm:

Users that are interested in modern-srwm are comparing it to the libraries listed below

• bhoov / energy-transformer-jax
  The Energy Transformer block, in JAX
  ☆56Updated last year
• glassroom / heinsen_sequence
  Code implementing "Efficient Parallelization of a Ubiquitious Sequential Computation" (Heinsen, 2023)
  ☆92Updated 4 months ago
• shyamsn97 / hyper-nn
  Easy Hypernetworks in Pytorch and Jax
  ☆100Updated 2 years ago
• bhoov / hamux
  Hierarchical Associative Memory User Experience
  ☆101Updated last year
• google-deepmind / synjax
  ☆246Updated 6 months ago
• ischlag / fast-weight-transformers
  Official code repository of the paper Linear Transformers Are Secretly Fast Weight Programmers.
  ☆104Updated 3 years ago
• DSL-Lab / GRBM
  Gaussian-Bernoulli Restricted Boltzmann Machines
  ☆104Updated 2 years ago
• volotat / ARC-Game
  The Abstraction and Reasoning Corpus made into a web game
  ☆89Updated 7 months ago
• akandykeller / NeuralWaveMachines
  Official Implementation of the ICML 2023 paper: "Neural Wave Machines: Learning Spatiotemporally Structured Representations with Locally …
  ☆70Updated last year
• glassroom / heinsen_routing
  Reference implementation of "An Algorithm for Routing Vectors in Sequences" (Heinsen, 2022) and "An Algorithm for Routing Capsules in All…
  ☆169Updated last year
• lucidrains / ponder-transformer
  Implementation of a Transformer that Ponders, using the scheme from the PonderNet paper
  ☆80Updated 3 years ago
• rasmusbergpalm / vnca
  ☆50Updated 2 years ago
• jxbz / agd
  Automatic gradient descent
  ☆207Updated last year
• lucidrains / Mega-pytorch
  Implementation of Mega, the Single-head Attention with Multi-headed EMA architecture that currently holds SOTA on Long Range Arena
  ☆203Updated last year
• louiskirsch / vsml-neurips2021
  Code for "Meta Learning Backpropagation And Improving It" @ NeurIPS 2021 https://arxiv.org/abs/2012.14905
  ☆31Updated 3 years ago
• GallagherCommaJack / modulax
  ☆17Updated 7 months ago
• HEmile / storchastic
  Stochastic Automatic Differentiation library for PyTorch.
  ☆198Updated 7 months ago
• google-deepmind / neural_networks_chomsky_hierarchy
  Neural Networks and the Chomsky Hierarchy
  ☆205Updated last year
• kingoflolz / swarm-jax
  Swarm training framework using Haiku + JAX + Ray for layer parallel transformer language models on unreliable, heterogeneous nodes
  ☆238Updated last year
• ag1988 / dss
  Sequence Modeling with Structured State Spaces
  ☆63Updated 2 years ago
• lixilinx / psgd_torch
  Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…
  ☆171Updated this week
• IDSIA / recurrent-fwp
  Official repository for the paper "Going Beyond Linear Transformers with Recurrent Fast Weight Programmers" (NeurIPS 2021)
  ☆48Updated 2 years ago
• ludwigwinkler / JaxLightning
  Running Jax in PyTorch Lightning
  ☆93Updated 3 months ago
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆83Updated last year
• dylandoblar / noether-networks
  Meta-learning inductive biases in the form of useful conserved quantities.
  ☆37Updated 2 years ago
• google-deepmind / physical_concepts
  ☆103Updated 3 years ago
• google-deepmind / neural_testbed
  ☆192Updated 10 months ago
• uclnlp / torch-imle
  Implicit MLE: Backpropagating Through Discrete Exponential Family Distributions
  ☆258Updated last year
• lucidrains / PaLM-jax
  Implementation of the specific Transformer architecture from PaLM - Scaling Language Modeling with Pathways - in Jax (Equinox framework)
  ☆187Updated 2 years ago
• maxencefaldor / cax
  Cellular Automata Accelerated in JAX (Oral at ICLR 2025)
  ☆84Updated last week