johnlyzhou Goto Github PK

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.

diayn-pytorch

Diversity is All You Need: Learning Skills without a Reward Function in PyTorch.

emgdecomp

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

melting-pot-contest-2023

model-free-opponent-shaping

Code for Model-Free Opponent Shaping (ICML 2022)

vame

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