This repository provides a PyTorch-based framework for medical image segmentation supporting both 2D and 3D architectures. It implements common segmentation algorithms like UNet and variants, supporting formats such as NIfTI and NRRD through the torchio library. The project includes training and inference code with configurable hyperparameters for multi-category segmentation tasks.