Hi Sebastian,
Can you please explain why these two distributions need to be reinstantiated on each forward pass?
My understanding was that the class definition doesnt execute any code that the graph depends on. The kl method does the forward pass. Why does class initiatilsation matter?