def onp_fun(x):
    out = onp.var(x.astype(jnp.promote_types(onp.float32, dtype)), axis=axis, ddof=ddof, keepdims=keepdims)
    return out.astype(out_dtype)
def onp_fun(x):
    out = onp.var(x.astype(jnp.promote_types(onp.float32, dtype)), axis=axis, ddof=ddof, keepdims=keepdims)
    return out.astype(out_dtype)