If you haven't already, take a look at Mirror - open source reimplementation of UNET with hundreds of bug fixes and improvements
Based on a superficial look, I'm worried it might have been more concerned with adding features than fixing and simplifying the fundamentals, but I don't really know.