GPU Infrastructure

[kogolobo]

We need to run our experiments on a GPU, so we need to update our minimal codebase to run on the department's compute.

[damienbeecroft]

We are using Google Colab as a common compute platform.