from AK_SSL.models.modules.heads import *

__all__ = [
    "SimCLRProjectionHead",
    "BarlowTwinsProjectionHead",
    "BYOLProjectionHead",
    "BYOLPredictionHead",
    "SimSiamProjectionHead",
    "SimSiamPredictionHead",
    "SwAVProjectionHead",
    "DINOProjectionHead",
]
