Use jax.Array instead of jax.typing.ArrayLike across the codebase. - #1728
Open
copybara-service[bot] wants to merge 1 commit into
Open
Use jax.Array instead of jax.typing.ArrayLike across the codebase.#1728copybara-service[bot] wants to merge 1 commit into
copybara-service[bot] wants to merge 1 commit into