Diagram of the network architecture

[DJayalath]

Is there a diagram of the neural network architecture (specifically the branches and heads) used in this paper?

[antonilo]

Yes, we have one in the paper in Fig. 6. (https://rpg.ifi.uzh.ch/docs/Loquercio21_Science.pdf)