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.