"vscode:/vscode.git/clone" did not exist on "8c025369ef5a9ccf89e416f07048cebfa7be4485"
-
Roy Hvaara authored
`jnp.array` is a function, not a type: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.array.html so it never makes sense to use `jnp.array` in a type annotation. Presumably the intent was to write `jnp.ndarray` aka `jax.Array`. Co-authored-by:
Peter Hawkins <phawkins@google.com>
fc639143