On a failed solve, check whether the input was NaN as well

[patrick-kidger]

Sometimes we detect that the output was NaN, and fail the linear solve. Under these circumstances we should check whether the input had any NaNs as well -- indicating that the culprit was probably something upstream.

Inspired by https://github.com/patrick-kidger/diffrax/issues/368.