Task-Specific-VAE Code for the MICCAI submission, available at Arxiv under: https://arxiv.org/abs/2304.06476