[super-pr] Polish lambda ray.sub and upstream

There's another ray.sub used to get around an issue with pyxis on one internal cluster, which can be upstreamed if cleaned up. This ray.sub gets around an issue where `--container-name ray-head` will actually create a new container even if it's overlapped which prevents us from overlapping the driver command on the head container.

This is a WAR, but this issue tracks the upstreaming of this so we don't have to maintain two ray.subs

cc @guyueh1