image-reconstruction neuroscience pytorch medical-imaging neuroimaging nsd fmri representation-learning clip uncertainty-estimation probabilistic-modeling multimodal-learning self-supervised-learning brain-decoding contrastive-learning stable-diffusion openclip retrieval-evaluation vision-reconstruction