[Core feature] [plugin] Type - Model analysis using WeightWatcher

[kumare3]

Motivation: Why do you think this is important?

For trained models, using WeightWatcher to analyze a deep learning model can help provide default analysis for a model

https://github.com/CalculatedContent/WeightWatcher

Goal: What should the final outcome look like, ideally?

Note: this is a proposal only. May not be the ideal implementation

@task(enable_flytedecks=True)
def train(...) -> Annotated[nn.Module, WeightWatcherRenderer]:
   ...

Describe alternatives you've considered

self publishing in custom deck.

Propose: Link/Inline OR Additional context

No response

Are you sure this issue hasn't been raised already?

• [X] Yes

Have you read the Code of Conduct?

• [X] Yes

Onboarding

• Go through getting started
• Go through visualizing artifacts
• Example of custom renderer
• Example of flytekit plugin with renderers

[ananyapillai72]

Would like to work on this

[github-actions[bot]]

Hello 👋, this issue has been inactive for over 9 months. To help maintain a clean and focused backlog, we'll be marking this issue as stale and will engage on it to decide if it is still applicable. Thank you for your contribution and understanding! 🙏

[github-actions[bot]]

Hello 👋, This issue has been inactive for over 9 months and hasn't received any updates since it was marked as stale. We'll be closing this issue for now, but if you believe this issue is still relevant, please feel free to reopen it. Thank you for your contribution and understanding! 🙏