Set Transformer
A simple neural network module for relational reasoning