Can dtypes of later arguments be annotated to follow the dtypes of earlier arguments?

Consider the function

def fill_array(array: Shaped[Array, "N"], fill_value: Any) -> Shaped[Array, "N"]:
    return jnp.full_like(array, fill_value)

In principle, the dtypes of fill_value and the return value should be constrained to be same as that of the input array. E.g., if I pass in an array of Integers, I should not specify a Float as the fill_value. Is it possible to annotate this constraint?

The issue names the fill_array example and jnp.full_like, but no repository files or tests. Start by tracing how jaxtyping expresses array dtypes and whether annotations can relate fill_value and the return value to the input array. Done means establishing whether this constraint is supported and, if so, adding the corresponding implementation and coverage.