['pytorch', 'safetensors', 'cceVAE', 'medical', 'anomaly detection', 'VAE', 'ceVAE', 'Compact ceVAE', 'unsupervised learning', '2D', 'StRegA', 'tumour segmentation', 'T1w', 'T2w', 'MRI', '3T', 'BraTS', 'MOOD', 'IXI', 'brain MRI', 'anomaly', 'image-segmentation', 'custom_code']