An implementation of (Induced) Set Attention Block, from the Set Transformers paper - View it on GitHub