lightly-ai/lightly

Implements modular building blocks including loss functions and model heads that expose SSL architecture components for customization. Supports 15+ SSL methods (BYOL, SimCLR, DINO, DINOv2, Barlow Twins, and others) with PyTorch and PyTorch Lightning backends, enabling distributed training across multiple GPUs. Integrates tightly with PyTorch ecosystems and includes example notebooks for rapid prototyping and production deployment.

3,697 stars and 236,532 monthly downloads. Used by 4 other packages. Actively maintained with 4 commits in the last 30 days. Available on PyPI.