code: def optimizer(self) -> Union[mx.optimizer.Optimizer, SockeyeOptimizer]:
          
          # TODO: Push update to MXNet to expose the optimizer (Module should have a get_optimizer method)
          return self.current_module._optimizer
docs: Returns the optimizer of the underlying module.