Reviewer # 1

Neural Information Processing Systems 

We thank the reviewer for the constructive feedback. We will make the suggested clarifications and fix the typos. The framework of the paper uses the model to improve the reparameterization directly. Reparameterizing in such an extension is an interesting future direction to explore. We thank the reviewer for the constructive feedback.