shibhansh / loss-of-plasticity

Related projects ⓘ

Alternatives and complementary repositories for loss-of-plasticity

• luchris429 / popjaxrl
  Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]
  ☆87Updated 11 months ago
• lindermanlab / S5
  ☆261Updated 3 months ago
• lucidrains / soft-moe-pytorch
  Implementation of Soft MoE, proposed by Brain's Vision team, in Pytorch
  ☆247Updated 6 months ago
• HazyResearch / hippo-code
  ☆167Updated 5 months ago
• Johswald / awesome-hypernetworks
  ☆58Updated 2 years ago
• JacksonWuxs / Forward-Forward-Network
  Implementation of Forward Forward Network proposed by Hinton in NIPS 2022.
  ☆162Updated last year
• shyamsn97 / hyper-nn
  Easy Hypernetworks in Pytorch and Jax
  ☆96Updated last year
• i404788 / s5-pytorch
  Pytorch implementation of Simplified Structured State-Spaces for Sequence Modeling (S5)
  ☆64Updated 6 months ago
• nnaisense / bayesian-flow-networks
  This is the official code release for Bayesian Flow Networks.
  ☆256Updated 4 months ago
• vivekmyers / contrastive_planning
  Code for the paper "Inference via Interpolation: Contrastive Representations Provably Enable Planning and Inference"
  ☆37Updated 4 months ago
• TariqAHassan / S4Torch
  PyTorch implementation of Structured State Space for Sequence Modeling (S4), based on Annotated S4.
  ☆70Updated 8 months ago
• YuhangSong / Prospective-Configuration
  This repository contains the code for the paper "Inferring Neural Activity Before Plasticity: A Foundation for Learning Beyond Backpropag…
  ☆80Updated 10 months ago
• lucidrains / minGRU-pytorch
  Implementation of the proposed minGRU in Pytorch
  ☆246Updated last week
• lucidrains / st-moe-pytorch
  Implementation of ST-Moe, the latest incarnation of MoE after years of research at Brain, in Pytorch
  ☆293Updated 5 months ago
• MadryLab / modelcomponents
  Decomposing and Editing Predictions by Modeling Model Computation
  ☆124Updated 5 months ago
• chrhenning / hypnettorch
  Package for working with hypernetworks in PyTorch.
  ☆117Updated last year
• GFNOrg / torchgfn
  A modular, easy to extend GFlowNet library
  ☆233Updated this week
• KindXiaoming / BIMT
  Brain-Inspired Modular Training (BIMT), a method for making neural networks more modular and interpretable.
  ☆162Updated last year
• PeaBrane / mamba-tiny
  Simple, minimal implementation of the Mamba SSM in one pytorch file. Using logcumsumexp (Heisen sequence).
  ☆102Updated last month
• AllanYangZhou / nfn
  NF-Layers for constructing neural functionals.
  ☆75Updated 10 months ago
• radarFudan / Awesome-state-space-models
  Collection of papers on state-space models
  ☆556Updated 2 weeks ago
• yilundu / irem_code_release
  ICML 2022: Learning Iterative Reasoning through Energy Minimization
  ☆43Updated last year
• AllanYangZhou / universal_neural_functional
  ☆48Updated 9 months ago
• Zhangyanbo / MLP-KAN
  Kolmogorov–Arnold Networks with modified activation (using MLP to represent the activation)
  ☆103Updated 3 weeks ago
• mkofinas / neural-graphs
  Official source code for "Graph Neural Networks for Learning Equivariant Representations of Neural Networks". In ICLR 2024 (oral).
  ☆70Updated 3 months ago
• metaopt / torchopt
  TorchOpt is an efficient library for differentiable optimization built upon PyTorch.
  ☆545Updated 2 weeks ago
• masa-ue / RLfinetuning_Diffusion_Bioseq
  Code for the tutorial/review paper for RL-based-fine-tuniing. In this code, we especially focus on the design of biological sequences li…
  ☆77Updated 2 months ago
• soniajoseph / ViT-Prisma
  ViT Prisma is a mechanistic interpretability library for Vision Transformers (ViTs).
  ☆179Updated this week
• keraJLi / rejax
  ☆149Updated this week
• NicolasZucchet / minimal-LRU
  Non official implementation of the Linear Recurrent Unit (LRU, Orvieto et al. 2023)
  ☆47Updated 2 weeks ago