What GPUs can I train on?

With Training Jobs, you declare the hardware your job needs in your Truss config, and Baseten provisions it — see the getting started guide for the config format. Available GPU configurations depend on your account; for large multi-GPU or multi-node runs, tell us what you're planning and we'll confirm capacity.

With Loops, Baseten picks the hardware for the base model automatically — you don't configure GPUs at all.

Docs: Training concepts · Multinode