jax

5cfd0e9a - Merge pull request #4441 from jakevdp:choice