Modularization via Optimal Transport