Fix dtype type annotations to use jax.typing.DTypeLike.#129
Open
Jammf wants to merge 1 commit intogoogle-deepmind:mainfrom
Open
Fix dtype type annotations to use jax.typing.DTypeLike.#129Jammf wants to merge 1 commit intogoogle-deepmind:mainfrom
jax.typing.DTypeLike.#129Jammf wants to merge 1 commit intogoogle-deepmind:mainfrom
