jax

25479c35 - fix pmap performance bug, dont always copy to host