jax

fee272e5 - Remove internal KeyArray alias

Remove internal KeyArray alias This was useful during the transition to typed PRNG keys, but is no longer necessary. It also makes generated HTML docs confusing: it's better to just use Array as we expect users to.