#PyTorch Geometric

Logo of DIG
DIG
DIG is a robust library designed to enhance the field of graph deep learning research, offering unified implementations and extensible frameworks. It stands out by supporting advanced research tasks like graph generation, self-supervised learning, and 3D graph analysis. DIG allows researchers to innovate and evaluate new methods alongside established ones using standard datasets and metrics. With the latest update built on PyTorch Geometric, it provides tools for diverse applications including explainability and graph augmentation, making it suitable for exploring graph neural networks and out-of-distribution issues.
Logo of torch-points3d
torch-points3d
This open-source framework leverages PyTorch Geometric and Facebook Hydra to facilitate the deployment of deep learning models on 3D point cloud data. It supports model development with features like mixed precision training, high-level API access, and methods including PointNet and PVCNN for tasks such as segmentation and object detection. The framework offers a streamlined setup process with Docker and supports diverse datasets such as ScanNet and ModelNet.