Issue when combining reparametrization and automatic enumeration

Misc.
3

Hi @fehiepsi,

Thanks a lot for your answer. It works indeed! So I guess there is a problem with LocScaleReparam.

Also, as noted here, the performance of hierarchical_dawid_skene is extremely slower than the numpyro version (>5mins vs 30sec). Perhaps the manual parametrization causes the gap in performance, but it is really surprising for me.

Anyway, thank you very much - I hope this was useful :slight_smile:

Best,
Pietro