liesel
liesel copied to clipboard
Update Array type annotations to use jax.Array instead of typing.Any
As the title says.