Add option for mutex timeout in distributed optimizer backward hook

Add option for mutex timeout in distributed optimizer backward hook #493