I'll take a stab at adding this new shared-private model in the toolbox! https://github.com/kasparmartens/shared-private-multimodalVAE https://arxiv.org/abs/2403.06338