KeyArray# gpjax.typing.KeyArray = jaxtyping.UInt32[jaxlib._jax.Array, '2'] | jaxtyping.Key[jaxlib._jax.Array, '']# Represent a PEP 604 union type E.g. for int | str