An interactive version of this notebook is available at this Google Colab link.