flax
2fbeb3b5 - Fix or ignore some pytype errors related to jnp.ndarray == jax.Array.

Commit

2 years ago

Fix or ignore some pytype errors related to jnp.ndarray == jax.Array. PiperOrigin-RevId: 511426098

Author

hawkinsp

Committer

a-googler

Parents