Official Github repository for the VAESim: A probabilistic approach for self-supervised prototype discovery paper