Format docstrings in training_algorithms/sgd.py