Add dSprites dataset

[FrankBrongers]

Add the dSprites dataset (https://github.com/deepmind/dsprites-dataset) or a reference or downloading script for it to the repo.

Align it with the rest of the code so experiments are easy to initiate (a dataloader or something similar)