Fix places that were passing jnp.array where a shape was expected.
For reference: https://github.com/google/jax/pull/6400.
For easier incorporation in open source projects, especially if they try to be reuse compliant, it would be great if the formats that support comments would contain license information.
The reuse specification recommends for that in markdown the following simple comment:
<!-- SPDX-FileCopyrightText: 2014 Coraline Ada Ehmke SPDX-FileCopyrightText: ... SPDX-FileCopyrightText: ... SPDX-License-Identifier: CC-BY-4.0 -->
commit sha： 61e4b9f4625287552ae606cd1367e48132cafbfdpush time in 17 hours ago