[Examples] Move examples to slapo.model_schedule

[chhzh123]

Description

This PR moves the schedules in the example models to slapo.model_schedule, so users can easily retrieve the scheduled models.

Checklist

• PR's title starts with a category (e.g. [Bugfix], [Model], [Tutorial], etc)
• Changes are complete (i.e. I finished coding on this PR)
• All changes have test coverage
• Code is well-documented

[chhzh123]

The epoi is removed? It seems the test failed because of the package dependency. @comaniac

[comaniac]

Otherwise LGTM. We could later design an API to dispatch the right "schedule_model" function based on the given model.

[comaniac]

The epoi is removed? It seems the test failed because of the package dependency. @comaniac

It should still be there or unit tests will fail. I'm not sure why there's a lint error tho...

[comaniac]

LGTM

[chhzh123]

I'm still changing the API. Will let you know when I'm done. @comaniac

[chhzh123]

Added apply_schedule(model, model_config, sch_config) API.

[comaniac]

We probably need an offline discussion on Monday about the APIs.

[chhzh123]

I've refactored all the code and changed the API to apply_schedule(model, schedule_key, **schedule_config). Should be ready to review. @comaniac

[comaniac]

Should be the last batch of comments.

[comaniac]

Per offline discussion this PR still needs to fix GPT-2 schedule.

[comaniac]

@chhzh123 I'll leave this PR to you for the final check and merge. Thanks.

[chhzh123]

I've tested. Thanks @comaniac.