code: def setup(self):
      self.encoder = Encoder(self.latents)
      self.decoder = Decoder()
  
file_path: jax/jax/experimental/jax2tf/tests/flax_models/vae.py
imports: from flax import linen
  import jax.numpy
  from jax import random
label_desc: Use image node_id 1 for calling the VAE obj's underlying member method code with example usage: obj.setup() without return types