internal_change

PiperOrigin-RevId: 600564077
google · Jan 22, 2024 · 00922cb · 00922cb
1 parent a909b9b
commit 00922cb
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/rax/utils.py b/rax/utils.py
@@ -14,6 +14,7 @@
 
 """Rax-specific utilities."""
 
+from rax._src.losses import pairwise_loss
 from rax._src.utils import approx_cutoff
 from rax._src.utils import approx_ranks
 from rax._src.utils import compute_pairs
@@ -25,5 +26,6 @@
     "approx_ranks",
     "compute_pairs",
     "cutoff",
+    "pairwise_loss",
     "ranks",
 ]