The beauty here is that the courts are separate from the government. The government can not in any way shape or form. Influence the courts in who or what they prosecute.

If they prosecuted a US soldier. There is nothing the Swedish government could do.

There is no special exemption. A deal was made. It includes no get out of jail free card for soldiers who brake the law.