Peter Hawkins 3082109a59 Add a type stub for jax.numpy.
This type stub is intended to match what pytype currently infers for jax.numpy, which is not particularly accurate in many cases. Future changes will add more accurate types to this stub.

Fix a number of new type errors this reveals to mypy.

PiperOrigin-RevId: 559179804
2023-08-22 11:50:49 -07:00
..
2023-08-22 11:50:49 -07:00

This sub-directory contains third-party code for which Google does not have copyright. Each sub-directory should correspond to a third-party library and must contain the appropriate LICENSE file. See instructions.