jax

02f65bb1 - Update warning message for jit of pmap