jax

0dbbc27b - Clarify that `grad` requires arguments to be differentiated to be of inexact type. (#2712)