Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add more AI/ML Training Examples #2040

Open
4 of 8 tasks
andreyvelich opened this issue Mar 29, 2024 · 8 comments
Open
4 of 8 tasks

Add more AI/ML Training Examples #2040

andreyvelich opened this issue Mar 29, 2024 · 8 comments

Comments

@andreyvelich
Copy link
Member

andreyvelich commented Mar 29, 2024

As we discussed previously: #2021 (comment) we want to add more AI/ML examples to the Kubeflow Training Operator. Right now, most of our examples have very basic and simple CNN training for MNIST. Since Training Operator is capable to train large-scale ML models, we would like to contribute more AI/ML use-cases.

We can make these examples Data Scientists friendly and re-use our Python SDK within Jupyter Notebooks to simplify the user submission.
I like the example structure of HF Transformers, so I propose the following path: examples/<framework>/<ml-use-case>

We can start with these examples (feel free to add more ML use-cases in this issue):

  • Language Modeling
  • Image Classification
  • Text Classification
  • Audio Classification
  • Question Answering
  • Speech Recognition
  • Text Generation
  • FSDP Example with PyTorch

We should investigate how to configure our CI/CD to make sure that these examples are functional.

cc @kuizhiqing @johnugeorge @tenzen-y @kubeflow/wg-training-leads

/help
/good-first-issue
/area example

Copy link

@andreyvelich:
This request has been marked as suitable for new contributors.

Please ensure the request meets the requirements listed here.

If this request no longer meets these requirements, the label can be removed
by commenting with the /remove-good-first-issue command.

In response to this:

As we discussed previously: #2021 (comment) we want to add more AI/ML examples to the Kubeflow Training Operator. Right now, most of our examples have very basic and simple CNN training for MNIST. Since Training Operator is capable to train large-scale ML models, we would like to contribute more AI/ML use-cases.

We can make these examples Data Scientists friendly and re-use our Python SDK within Jupyter Notebooks to simplify the user submission.
I like the example structure of HF Transformers, so I propose the following path: examples/<framework>/<ml-use-case>

We can start with these examples (feel free to add more ML use-cases in this issue):

  • Language Modeling
  • Image Classification
  • Text Classification
  • Audio Classification
  • Question Answering
  • Speech Recognition
  • Text Generation

We should investigate how to configure our CI/CD to make sure that these examples are functional.

cc @kuizhiqing @johnugeorge @tenzen-y

/help
/good-first-issue
/area example

Instructions for interacting with me using PR comments are available here. If you have questions or suggestions related to my behavior, please file an issue against the kubernetes/test-infra repository.

@xr-dev-saurabh
Copy link

/assign

@StefanoFioravanzo
Copy link
Member

@andreyvelich I love this. Few thoughts:

  • Whenever we publish a new example, please reach out to me or Amber so that we can help turning it into either a short blog post or at least disseminate via social media.
  • How do you define the actual use case for these topics?
  • Are these examples supposed to be specific to the training operator or were you thinking of a wider applicability (serving, tuning, metadata, etc.)

Copy link

github-actions bot commented Jul 9, 2024

This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions.

@andreyvelich
Copy link
Member Author

/remove-lifecycle stale

Copy link

github-actions bot commented Oct 9, 2024

This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions.

@andreyvelich
Copy link
Member Author

/remove-lifecycle stale

@snax-07
Copy link

snax-07 commented Nov 10, 2024

/assign

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

No branches or pull requests

4 participants