https://github.com/deepmind/dsprites-dataset