PyTorch-based healthcare imaging framework