Install with pip install -e . from this directory.
Use with import distdl_unet.
Does not require DistDL to use classical UNets.
See examples directory for an example with the classical UNet.
Looking at a dummy UNet's modules can help visualize the network itself.