egg/core/losses.py
egg/zoo/emcom_as_ssl/losses.py

egg/core/gs_wrappers.py
egg/zoo/emcom_as_ssl/archs.py

egg/core/continous_communication.py
egg/zoo/emcom_as_ssl/archs.py