from .layerwise_amalgamation import LayerWiseAmalgamator