I want to use AI coding agents for machine learning projects [D]

I'm a software engineer who mainly builds softwaes/applications, and I'm starting to work on machine learning projects.

Since ML workloads often require GPUs, I know services like Google Colab and Kaggle exist. but, I'm looking for something a bit different.

Is there a platform where I can use AI coding agents (such as Codex, Claude Code, or OpenCode) while running the actual ML code on a cloud GPU?

Ideally, I'd like to:

• Work locally with my preferred editor and AI coding agent.
• Have the code execute on a remote GPU machine.
• Be able to build, debug, and iterate on ML projects as if the GPU were attached to my local development environment.

Does a setup like this exist? If so, what tools or platforms do you recommend?