When does one prefer to use Trace_ELBO() and TraceEnum_ELBO() . Under what conditions or constraints ? It would be very helpful. Thanks!!
Use Trace_ELBO or TraceGraph_ELBO when you want to use control variates to reduce variance in gradient estimators for non-reparametrizable distributions (e.g. discrete distributions when enumeration is intractable, as in Attend-Infer-Repeat). Use TraceEnum_ELBO when you want to integrate out discrete variables in your model or guide with enumeration.