Contrastive Learning of Structured World Models - View it on GitHub