I notice that you use `GatedDense` in `self.q_z_layers` to encode `q(z | x)`:  Is there any reason to use GatedDense rather than a simple MLP? Thanks!
I notice that you use
GatedDenseinself.q_z_layersto encodeq(z | x):Is there any reason to use GatedDense rather than a simple MLP? Thanks!