Fig. 3: Node features. Encoder The Encoder processes particle positions into relevant features in a GraphTensor structure and embeds them into the tfgnn.HIDDEN_STATE vector. Windowing We start with ...