feat: support for managing gpu enabled self runner infra #1436

Job	Run time
Build and Publish Images (deepspeed-runtime, cmd/runtimes/deepspeed/Dockerfile, linux/amd64,linux...	39m 11s
Build and Publish Images (model-initializer, cmd/initializers/model/Dockerfile, linux/amd64,linux...	1m 56s
Build and Publish Images (trainer-controller-manager, cmd/trainer-controller-manager/Dockerfile, ...	6m 38s
Build and Publish Images (mlx-runtime, cmd/runtimes/mlx/Dockerfile, linux/amd64)	26m 22s
Build and Publish Images (torchtune-trainer, cmd/trainers/torchtune/Dockerfile, linux/amd64,linux...	15m 28s
Build and Publish Images (dataset-initializer, cmd/initializers/dataset/Dockerfile, linux/amd64,l...	1m 57s
	1h 31m 32s

Provide feedback