Implement FactorVAE

[Frank Brongers]

Implement the FactorVAE and make it functional within the training and testing loop with the dSprites dataset.

[Frank Brongers]

Continuing with issue on https://github.com/FrankBrongers/FactorVAE