@vjevtic @chasse @cprouve @mfuehrin implementation of deepsets in the tagging infrastructure
cc @vchobano @bkhanji
gives the same results as the pytorch implementation: