jax

810b6f82 - Use Tuple[int, ...] rather than Sequence[int] in jnp.ndarray shape annotation.