Add NumPy Stubs
We use ndarray often in our annotations, but it always just resolves to `Any` since it doesn't have any annotations natively. It looks like numpy has a separate Numpy-stubs repo under development:

https://github.com/numpy/numpy-stubs

Might be worth considering that as a development dependency. Certainly would like to find issues with ndarray usage sooner than later

@shoyer 