An unofficial implementation of the Set Transformer framework in Tensorflow/Keras based on the paper: Set Transformer: A Framework for Attention-based Permutation-Invariant Neural Networks.
The official pytorch repository is available at: https://github.com/juho-lee/set_transformer.
The set transformer framework can be installed by executing the following command:
pip install tf-settransformer