Task Scheduling for Single node in DL/RL scheduling

I was wondering, does anyone have a resource or code repo for build or create a single node level Scheduler for deep learning or machine learning based RLScheduler for one task scheduling. I mean using python or Pytorch, how can I create a single node task Scheduler based on machine learning or deep learning model into the Cluster. I appreciate if anyone could help me regarding this. Thanks