pyg-team / pytorch_geometric

We write your reusable computer vision tools. 💜

Implement a ChatGPT-like LLM in PyTorch from scratch, step by step

🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and training.

Ultralytics YOLO27, YOLO26, YOLO11, YOLOv8 — object detection, instance segmentation, semantic segmentation, image classification, pose estimation, object tracking