Simple normalizing flow with a conditional variable of any size - View it on GitHub