jax

c100eb66 - Merge pull request #33705 from jakevdp:del-jax-metal