/attention-neuron

Re-implementation of https://ai.googleblog.com/2021/11/permutation-invariant-neural-networks.html

Primary LanguageJupyter Notebook

Stargazers