Have some improvements to the Ray plugin:
https://github.com/flyteorg/flyte/pull/5933 CC:
@glamorous-carpet-83516
One thing I'm wondering is how we're supposed to handle changes to affect the flytekit via flyteidl. Do we just wait until flyteidl releases and then update flytekit?