Wow, nice repo. The CSS.ipynb notebook is super clear and easy to follow, almost easier to understand than reading the paper.