← all repositories

Deci-AI/super-gradients

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

5k stars Jupyter Notebook ML FrameworksComputer Vision
super-gradients
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.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.