https://www.arxiv-summary.com/posts/2210.05666/
Point Transformer V2: Grouped Vector Attention and Partition-based Pooling