Add support for (single-layer) weighted graphs

[ragibson]

It would be nice to officially add support for weighted graphs even though this is outside the scope of our paper. Several parts of the pruning pipeline already handle edge weights and the extension in the single-layer case should be relatively straightforward.