diff --git a/src/moscot/problems/cross_modality/_translation.py b/src/moscot/problems/cross_modality/_translation.py
index 0e702bea4..5c8f9081c 100644
--- a/src/moscot/problems/cross_modality/_translation.py
+++ b/src/moscot/problems/cross_modality/_translation.py
@@ -226,6 +226,7 @@ def solve(  # type: ignore[override]
         threshold: float = 1e-3,
         linear_solver_kwargs: Mapping[str, Any] = types.MappingProxyType({}),
         device: Optional[Literal["cpu", "gpu", "tpu"]] = None,
+        **kwargs: Any,
     ) -> "TranslationProblem[K]":
         r"""Solve the translation problem.
 
@@ -303,6 +304,7 @@ def solve(  # type: ignore[override]
             threshold=threshold,
             linear_solver_kwargs=linear_solver_kwargs,
             device=device,
+            **kwargs,
         )  # type: ignore[return-value]
 
     @property