['pytorch', 'safetensors', 'ProbUNet', 'medical', 'probabilistic unet', '2D', 'PULASki', 'multiple sclerosis segmentation', 'multiple sclerosis', '3T FLAIR', 'FLAIR', 'MRI', '3T', 'multiple rater', 'Conditional VAE', 'distribution distance', 'image-segmentation', 'custom_code']