jax

c3aa8692 - Merge pull request #31994 from mattbahr:implement-hyp2f1-derivative