Hi @twiecki ,
Good day.
I manage to solve both issue.
I wait for a while and updated to Pymc3 v4, although it is still unstable now.
regarding the problem I mentioned in example, actually is my settings issue. If we install numpyro v0.6.0 and jax v0.2.10, the exmaple will run with no problem.
I will still wait for official release of Pymc3 v4.
Thank you very much.