def onp_fun(arg):
    arg = (arg.astype(onp.float32) if (dtype == jnp.bfloat16) else arg)
    return onp.vander(arg, N=n, increasing=increasing)
def onp_fun(arg):
    arg = (arg.astype(onp.float32) if (dtype == jnp.bfloat16) else arg)
    return onp.vander(arg, N=n, increasing=increasing)