Closed fritzo closed 2 years ago
@fehiepsi it looks like the jax tests may be failing due to numpyro renaming of distributions? could you ptal?
It looks like the issue is related to the recently added distribution RelaxedBernoulli. Let me tal.
Thanks for fixing the jax issues @fehiepsi! Would you be able to approve this PR?
pip install --update black && make format