quark.torch.pruning.model_transformation