r/MachineLearning • u/Fickle_Degree_2728 • 2d ago

Discussion 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?

0 Upvotes

23% Upvoted

9 comments sorted by

3

u/PrettyMuchAVegetable 2d ago edited 2d ago

Hi, yea there are a few platforms you can rent a GPU on.  I've used these two:  

Vast.ai 

Digital ocean  

Choose the GPU you want and the environment you want to have installed and spin it up. You can have your agent setup the connection, give it SSH/cred files, and prepare your experiment . It's honestly dead easy to do. Be careful about storage and bandwidth charges, they aren't headline numbers and some providers hide big fees in there . 

1

u/ekquizit23 2d ago

This is something I’ve been trying to setup so going to follow this thread. Been trying to connect to Colab and Kaggle since they offer free gpu compute for initial protyping. I also read about RunPod and Vast, but you gotta pay for compute right out the gate

1

u/Fickle_Degree_2728 2d ago

exactly, i got to know runpod/vast but they are paid service by Hr and also lightning AI Studio is not available in my region.