jax

40b9554d - Support `device_put` of an uncommitted array to a single global device in McJAX.