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 in this repository:
pip install .