redo all docs to match pytorch theme. Adjust all documentation to the new code structure. Remove partial imports from...everywhere. Updated examples