jax

f17d0f38 - Merge pull request #22664 from jakevdp:astype-device