jax

15009c90 - add docstring for defjvps, fix sphinx docs