Modules
tiny_diff.modules.attention
tiny_diff.modules.conv
tiny_diff.modules.layer_factory
tiny_diff.modules.nonlinearity
tiny_diff.modules.norm
tiny_diff.modules.resampling
tiny_diff.modules.residual