Contrary to what the docstring says, gradient descent does not accept a sequence for the `step_rate` parameter. I'm happy to submit a pull request for this, if there's still interest!