.. automodule:: torchft.local_sgd
    :members:
    :undoc-members:
    :show-inheritance: