jnp.linalg.cond: return inf rather than NaN for singular matrices

For p in {None, 2}, cond returned early and skipped the NaN->inf
conversion applied to every other norm, so an exactly singular input
gave 0/0 = NaN where NumPy (and cond itself with any other p) gives
inf.