* Add failed assertion message in CI

* Pin numpy upper bound in numba install

numba-scipy downgrades the installed scipy to 1.7.3 in Python 3.8, but not numpy, even though scipy 1.7 requires numpy<1.23. When installing PyTensor next, pip installs a lower version of numpy via the PyPI.

* Run numba and jax tests in separate jobs

---------
Co-authored-by: Ben Mares <services-git-throwaway1@tensorial.com>