smonsays

recurrentgemma

Open weights language model from Google DeepMind, based on Griffin.

minimal-hypernetwork

Minimal hypernetwork implementation.

modular-hyperteacher

Code accompanying the paper Discovering modular solutions that generalize compositionally

cola

Compositional Linear Algebra

machine-tasks

Datasets for compositional learning

metax

flexible meta-learning in jax

pushworld

PushWorld: A benchmark for manipulation planning with tools and movable obstacles

cocoa

Code accompanying the paper "Would I have gotten that reward? Long-term credit assignment by counterfactual contribution analysis"

groundedSCAN

Grounded SCAN data set.

shrink-perturb

Optax implementation of shrink and perturb (Ash & Adams, 2020).

netrep

Some methods for comparing network representations in deep learning and neuroscience.

S5

permuted_initialisations

Code accompanying "Random initialisations performing above chance and how to find them", for paper see https://arxiv.org/abs/2209.07509

amortized-optimization-tutorial

Tutorial on amortized optimization for learning to optimize over continuous domains

jax-hypernetwork

A simple hypernetwork implementation in jax using haiku.

flaxify

Convert haiku modules to flax

xplogger

Logging utility

gymnax

RL Environments in JAX 🌍

gymnax-blines

Baselines for gymnax 🤖

eprop-PyTorch

PyTorch implementation of the eligibility propagation (e-prop) learning algorithm.

evosax

Evolution Strategies in JAX 🦎

BinomialSynapses.jl

High-performance Julia package for online learning of binomial synaptic release models

minimal-meta-rl

Minimal A2C/A3C example of an LSTM-based meta-learner.

presynaptic-stochasticity

Presynaptic Stochasticity Improves Energy Efficiency and Alleviates the Stability-Plasticity Dilemma

gridtorch

PYTORCH Implementation of the supervised learning experiments in Vector-based navigation using grid-like representations in artificial agents, as published at https://www.nature.com/articles/s41586-018-0102-6

learning_where_to_learn

hypnettorch

Package for working with hypernetworks in PyTorch.

deep_feedback_control

Official code repository for the publication "Credit Assignment in Neural Networks through Deep Feedback Control".

randman

spytorch

Tutorial for surrogate gradient learning in spiking neural networks