kd.typing.PRNGKeyLike

[[Source]]

PRNGKeyLike = Union[int, Sequence[int], np.ndarray, PRNGKey]