Open-source search and retrieval database for AI applications.

Latex code for making neural networks diagrams

A simple Python debugger and profiler that generates animated visualizations of program flow, useful for algorithm learning.

Data Visualization Server

Open-source implementations of OpenAI Gym MuJoCo environments for use with the OpenAI Gym Reinforcement Learning Research Platform.

scalable analysis of images and time series

CALVIN - A benchmark for Language-Conditioned Policy Learning for Long-Horizon Robot Manipulation Tasks

CLIPort: What and Where Pathways for Robotic Manipulation

CLU lets you write beautiful training loops in JAX.

A CLI for processing composite Wavefront OBJ files for use in MuJoCo.

Mechanistic Interpretability for Transformer Models

Using deep learning to referee fencing

Environments to support https://github.com/sholtodouglas/learning_from_play and reinforcement learning for robotic manipulation.

Tensorflow Implementation of "Generating Videos with Scene Dynamics"

Minimalistic Google Docs based workflow for Distill.pub

Instructions on how to create an MPI cluster in gcloud

My attempt at implementing Deepmind's MuZero (https://arxiv.org/abs/1911.08265)

A easily modifiable version of SAC, HER, behavioural cloning and a few other algorithms I used for some experiments.