jax

ceca1b7a - try not to sabotage all jax users, use normalized platform names to control device_put behavior.