Fixed a series of breaking changes related to the metrics returning arrays. Adapted docs to the new functionalities

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