Stage 2: attention-head pruning results + mask + apply_mask.py

Initialize repo with Stage 0 baseline