Linking here since the pruning method you're talking about wasn't the top search result when I looked: https://github.com/CerebrasResearch/reap