jax

f1b14aa2 - Merge pull request #4820 from jakevdp:fix-type-promotion