jax

9f9cc269 - Merge pull request #29930 from jakevdp:device-doc