hey @mmisiur, I have a question. Since I am optimizing over both the networks, is it okay if there is a function that processes the output of the first network before feeding it to the next network? It did work in Tensorflow though. Thanks for your time.