NaN output for latent variable after the reparameterize

[Baran-phys]

For the set generation task, even with an MLP set generator, the latent variable gives NaN values since after the reparameterize trick, the max and min values of z are -inf and +inf.
What do you think?