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

Update isp to allow modeling single perturbation in multiple cells as batches

Add in silico perturbation example notebook.