Back to problems

Debug a Buggy UNet in PyTorch

AI Coding · Apple · Hard

Repair a Faulty PyTorch UNet ML Coding Phone Machine Learning Engineer Reported February, 2026 Problem Overview This Apple Machine Learning Engineer phone-screen exercise has caught candidates who had not studied the architecture beforehand. You receive an almost-correct PyTorch version of the UNet introduced by Ronneberger et al. in 2015, plus a required input/output specification. It either fails to execute or returns tensors with an incorrect shape, and you must correct…

Checking your access…