[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?

• Yes

Have you read the Code of Conduct?

• 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]

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! 🙏