masa-su / pixyz

A library for developing deep generative models in a more concise, intuitive and extendable way
https://pixyz.io
MIT License
487 stars 41 forks source link

Implementation of NVAE #131

Open yizhe-ang opened 3 years ago

yizhe-ang commented 3 years ago

Hi, I just came across this library and it looks awesome. Keep up the good work!

I was just wondering if a model such as the NVAE (https://arxiv.org/abs/2007.03898), i.e. a hierarchical VAE, with an inference and generative network that shares parameters with each other, will be possible to implement using the Pixyz framework.

Thank you in advance!