Stop using numpy.typing in favour of our types and lower numpy compat bound
Created by: PhilipVinc
To numpy>=1.18. In any case this is what jax/stax require so it makes no sense to go lower. It was reported to me it was annoying on google colab bcause tensor flow requires numpy<1.20