Deci-AI/super-gradients

Health v2 · maintenance only

Adoption not part of the score

5052 stars · 592 forks observed · 2026-08-28

What it is AI-extracted, prompt v1, taxonomy v1, 2026-08-29, confidence not recorded

SuperGradients is an open-source PyTorch-based training library for building, training, and fine-tuning state-of-the-art computer vision models, and is the home of the Yolo-NAS object detection model. It provides pretrained model checkpoints and recipes for classification, object detection, segmentation, and pose estimation tasks.

Use cases

• train a custom object detection model on my own dataset
• fine-tune Yolo-NAS for detection
• fine-tune a pretrained image classification model
• train a semantic segmentation model in PyTorch
• run pose estimation with a pretrained model
• export a vision model for production deployment

When to choose

• you want a single library covering classification, detection, segmentation, and pose estimation
• you want access to Yolo-NAS and other SOTA pretrained checkpoints
• you prefer recipe-driven PyTorch training with sensible defaults

When to avoid

• you need tasks outside computer vision such as NLP or audio
• you want a lightweight inference-only runtime rather than a training library
• you need a framework-agnostic solution beyond PyTorch

Facets

library · maturity active

machine-learning deep-learning computer-vision image-processing computer-vision deep-learning machine-learning image-processing python cross-platform yolo-nas object-detection image-classification semantic-segmentation pose-estimation model-training fine-tuning pretrained-models pytorch gpu

Member repositories

For agents

markdown · JSON · MCP: product_card(name="Deci-AI/super-gradients")