You might be accessing a different environment on spyder and on the notebook. You can print sys.executable to check if they match.
On another note your version of PyMC is quite old, I suggest you upgrade. In that case you can reach jax sampling via pm.sample(nuts_sampler="numpyro")