Libraries & frameworks
pytorch_geometric
pyg-team/pytorch_geometric
A PyTorch-based library for graph neural networks, offering GNN layers, pooling, models, and scalable training utilities.
Topics
Related projects
ultralytics63.3kA Python library for YOLO object detection, segmentation, pose, and tracking with training and deployment tools.
EasyOCR30.1kA ready-to-use Python OCR library supporting 80+ languages across Latin, Chinese, Arabic, and Cyrillic scripts.
onnx21.7kAn open standard and Python package for AI model interoperability, defining a common graph format across frameworks.
open_clip14.2kAn open-source CLIP implementation with training code and pretrained models for vision-language learning.
pytorch-grad-cam13kA PyTorch library of state-of-the-art visual explanation methods for diagnosing classification, detection, and segmentation models.
kornia12.3kA differentiable computer vision library built on PyTorch for image processing, geometric vision, and augmentation pipelines.