Yama is the god of justice, who also collects the dead. He is only a ‘punisher’ in the sense that he is a judge. He is usually shown as being happy to cancel or reduce punishments if the dead person can give some legal justification for their actions.

If you have read Discworld, Death is very similar to Yama, (except that Death leaves the judgement to the dead person).