Christina Theodoris
Fix isp perturb_group dims, reformat cell states dict to keyed, add attn mask
c2679c4