/set_transformer

A TensorFlow implementation of the paper 'Set Transformer: A Framework for Attention-based Permutation-Invariant Neural Networks'

Primary LanguageJupyter Notebook

Stargazers