PIMA_torch Pytorch implementation of physics informed multimodal autoencoders (PIMA), used to generate figures for paper