Created demo colab notebook, and link in readme#352

evelynmitchell wants to merge 2 commits intoprobml:mainprobml/dynamax:mainfrom

evelynmitchell:mainevelynmitchell/dynamax:mainCopy head branch name to clipboard