Deci-AI/super-gradients
A PyTorch-based training library for building, training, and fine-tuning production-ready SOTA computer vision models including Yolo-NAS.

Velocity · 7d
+3.0
★ / day
Trend
→steady
star history
SuperGradients is an open-source deep learning training library built on PyTorch. It provides unified tooling for training and fine-tuning state-of-the-art computer vision models including object detection architectures like Yolo-NAS. The library offers pretrained model weights, training recipes, and utilities for common computer vision tasks such as object detection, segmentation, and classification.