johnlyzhou

behavenet

Toolbox for analyzing behavioral videos and neural activity

behavenet_neurocaas

PS-VAE analysis implementation and hyperparameter search for NeuroCAAS

brax

Massively parallel rigidbody physics simulation on accelerator hardware. Modified to add an environment for motion imitation learning with a two-link arm.

decision-transformer

Official codebase for Decision Transformer: Reinforcement Learning via Sequence Modeling.

emgdecomp

Package for decomposing EMG signals into motor unit firings, as used in Formento et al 2021.

init-entropy

Init entropy

Melting-Pot-Contest-2023

model-free-opponent-shaping

Code for Model-Free Opponent Shaping (ICML 2022)

reciprocator

Code for "Reciprocal Reward Influence Encourages Cooperation From Self-Interested Agents" (2024)

VAME

Variational Animal Motion Embedding with modifications/additional methods for open field datasets