Upgrade to numpyro 0.4.0

Supports numpyro 0.4.0 and the jax updates therein! Also adds a check for null data in fit/predict and raises an exception that makes the problem obvious.