fix: use jnp.where for NaN-safe masking in vmapped_deflections_from

fix: use jnp.where for NaN-safe masking in vmapped_deflections_from #172