Thanks a lot for the pointers @madlag! I didn’t know about your pytorch_block_sparse repo - this is exactly what I’m looking for 
I’ll keep an eye out for the release of your tool - do I understand correctly that this will enable people to incorporate e.g. movement pruning in their workflow or is it more focused on patching networks?
Cheers,
Lewis