input + output pd data frames, adopt compute_feats_grads
stow gradient localization demo with other notebooks