 matter related to its
general usage and behavior. Inherent JAX features such as the following are supported: Just-In-Time (JIT) compilation Automatic Differentiation Vectorization Parallelization  FlaxAutoencoderKLOutput  class diffusers.models.vae_flax.FlaxAutoencoderKLOutput  < source > ( latent_dist: FlaxDiagonalGaussianDistribution  )   Parameters   latent_dist (FlaxDiagonalGaussianDistribution) —
Encoded outputs of Encoder represented as 