mbchang Goto Github PK

jax-iterative-inference

kmeans_pytorch

pytorch implementation of basic kmeans algorithm(lloyd method with forgy initialization) with gpu support

maddpg

Code for the MADDPG algorithm from the paper "Multi-Agent Actor-Critic for Mixed Cooperative-Competitive Environments"

magical

The MAGICAL benchmark suite for robust imitation learning (NeurIPS 2020)

markov-music

mbchang.github.io

mbrl-lib

Library for Model Based RL

mnist_transforms

multiagent-particle-envs

Code for a multi-agent particle environment used in the paper "Multi-Agent Actor-Critic for Mixed Cooperative-Competitive Environments"

multiworld

Multitask Environments for RL

nri

Neural relational inference for interacting systems - pytorch

o2p2

op3

pella-io

Go to

physics-worlds

pytorch-a2c-ppo-acktr-gail

PyTorch implementation of Advantage Actor Critic (A2C), Proximal Policy Optimization (PPO), Scalable trust-region method for deep reinforcement learning using Kronecker-factored approximation (ACKTR) and Generative Adversarial Imitation Learning (GAIL).

pytorch-rl

PyTorch implementation of Deep Reinforcement Learning: Policy Gradient methods (TRPO, PPO, A2C) and Generative Adversarial Imitation Learning (GAIL). Fast Fisher vector product TRPO.

pytorch-vqvae

Vector Quantized VAEs - PyTorch Implementation

ravens

Train robotic agents to learn pick and place with deep learning for vision-based manipulation in PyBullet. Transporter Nets, CoRL 2020.

relational-nem

Code for the "Relational Neural Expectation Maximization: Unsupervised Discovery of Objects and their Interactions" paper.

rl-starter-files

RL starter files in order to immediatly train, visualize and evaluate an agent without writing any line of code

rlkit

Collection of reinforcement learning algorithms

submitit

Python 3.6+ toolbox for submitting jobs to Slurm

torch-ac

Recurrent and multi-process PyTorch implementation of deep reinforcement Actor-Critic algorithms: A2C and PPO

torch7

http://torch.ch

video-annotation

Quickly annotate videos with bounding boxes for use in training object detectors and trackers.