* feat: add Dropout to Flax UNet * feat: add @compact decorator * fix: drop nn.compact
Attach a file by drag & drop or click to upload
