def onp_fun(arr, values):
    arr = (arr.astype(onp.float32) if (arr.dtype == jnp.bfloat16) else arr)
    values = (values.astype(onp.float32) if (values.dtype == jnp.bfloat16) else values)
    out = onp.append(arr, values, axis=axis)
    return out.astype(jnp.promote_types(*arg_dtypes))
def onp_fun(arr, values):
    arr = (arr.astype(onp.float32) if (arr.dtype == jnp.bfloat16) else arr)
    values = (values.astype(onp.float32) if (values.dtype == jnp.bfloat16) else values)
    out = onp.append(arr, values, axis=axis)
    return out.astype(jnp.promote_types(*arg_dtypes))