While many countries (especially Middle-eastern countries) abuse diplomatic immunity with regards to smaller crimes (and especially parking tickets), most countries choose not to exercise the immunity in the case of felonies (except where the death penalty is on the table). Alternatively, the country will exercise the immunity but will then honor an extradition request.
IOW, unless Assange gets himself appointed as a diplomat of Australia, he's done nothing to improve his situation in the US.