GeoT: Tensor Centric Library for Graph Neural Network via Efficient Segment Reduction on GPU

Zhongming Yu, Genghan Zhang, Hanxian Huang +2

In recent years, Graph Neural Networks (GNNs) have ignited a surge of innovation, significantly enhancing the processing of geometric data structures such as graphs, point clouds,…