I unfortunately didn't write code for this post (other than adapting the visualizations) - I mainly focused on the theoretical concepts in the paper.