You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Currently, it seems that the grad trick in numpyro.render_model does not work well with extreme values (like 1e-38) and gives wrong visualization (as in #1241 (comment))). This is another motivation for having ProvenanceTensor.
It would be nice to have a version of ProvenanceTensor to infer dependency structure of a numpyro model.
For now, it is not clear to me how to do the same thing in JAX. :(
The text was updated successfully, but these errors were encountered: