[ENHANCEMENT] Runtime and scheduler refactor so that we support uniform scheduler API

[huanggx-sea]

Done in https://github.com/sail-sg/zero-bubble-pipeline-parallelism/pull/39 and merged into dev branch.