jax

31695257 - Merge pull request #31133 from jakevdp:polygamma-doc