Hi @fehiepsi ,
Good day.
The error still not solve even I install ipywidgets.
But if we refer to this forum,
After I replace args = parser.parse_args() with
args, unknown = parser.parse_known_args()
The error solved.
In conclusion, installed ipywidgets + the small modification to the code solve the problem.
Hopefully you can update on the documentation.
Thank you very much.
Python 3.7.10 + miniconda(jupyter notebook)
numpyro v0.6.0
jax v0.2.10
jaxlib v0.1.62+cuda101