.. automodule:: satrain.pytorch
    :members: