Docs improvement for `Module.from_checkpoint`

[CharlelieLrt]

PhysicsNeMo Pull Request

Description

This PR improves the docstring for the method Module.from_checkpoint with added details regarding the mechanism to infer the class, as well as recommended usage.

Note: This PR has a companion PR in the documentation repo, and they should be merged together.

Checklist

• [x] I am familiar with the Contributing Guidelines.
• [x] New or existing tests cover these changes.
• [x] The documentation is up to date with these changes.
• [x] The CHANGELOG.md is up to date with these changes.
• [x] An issue is linked to this pull request.

Dependencies