def METHOD_NAME(atom14, batch):
    atom37_data = batched_gather(
        atom14,
        batch['residx_atom37_to_atom14'],
        dim=-2,
        num_batch_dims=len(atom14.shape[:-2]),
    )
    atom37_data = atom37_data * batch['atom37_atom_exists'][..., None]
    return atom37_data