--- a
+++ b/unimol/losses/__init__.py
@@ -0,0 +1,7 @@
+from pathlib import Path
+import importlib
+
+# automatically import any Python files in the criterions/ directory
+for file in sorted(Path(__file__).parent.glob("*.py")):
+    if not file.name.startswith("_"):
+        importlib.import_module("unimol.losses." + file.name[:-3])