Hello again,
When I use ADVI, I get ok results but still pretty far off the mark. The ELBO seems to stop improving pretty quickly:
When I try Householder flow, my kernal dies every time. Do you have any idea why that might be or how to improve it?
Thanks again!