Update docstrings

skrl/models/jax/base.py

@@ -527,7 +527,7 @@ def migrate(self,
                 name_map: Mapping[str, str] = {},
                 auto_mapping: bool = True,
                 verbose: bool = False) -> bool:
-        """Migrate the specified extrernal model's state dict to the current model
+        """Migrate the specified external model's state dict to the current model
 
         .. warning::
 
@@ -613,6 +613,9 @@ def reduce_parameters(self, tree: Any) -> Any:
         :param tree: pytree to apply collective reduction
         :type tree: Any
 
+        :return: all-reduced pytree
+        :rtype: Any
+
         Example::
 
             # reduce model parameter across all workers/processes