Open-source search and retrieval database for AI applications.

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

CLU lets you write beautiful training loops in JAX.

Minimalistic Google Docs based workflow for Distill.pub

scalable analysis of images and time series

Data Visualization Server

Mechanistic Interpretability for Transformer Models

Instructions on how to create an MPI cluster in gcloud

CLIPort: What and Where Pathways for Robotic Manipulation

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

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

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

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

Latex code for making neural networks diagrams

Using deep learning to referee fencing

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

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

Tensorflow Implementation of "Generating Videos with Scene Dynamics"