Support passing in a ParameterGroup for param_group arguments