Distributed AI Training & Robotics Infra Intern

Roboticsconnected; Santa Clara (CA); Presencial; USD 41.000 - 69.000; Tempo integral

Resumo da vaga

NVIDIA's Cosmos Lab Infrastructure team is seeking an intern to help develop training and post-training systems for Physical AI models, including world foundation models and robot policies. You’ll work with a mentor on a focused project and implement improvements on real AI workloads using NVIDIA's GPU infrastructure.

You will contribute to training, inference, simulation, and evaluation pipelines, collaborating with researchers to optimize performance and scalability.

Qualificações

• Pursuing a Bachelor's, Master's, or PhD in Computer Science, Computer Engineering, Electrical Engineering, or a related field.
• Strong Python and debugging skills, with systems fundamentals in concurrency and distributed execution.
• Experience in training infrastructure, RL infrastructure, or robotics integration is valued.

Responsabilidades

• Develop and optimize training infrastructure for advanced Physical AI world models.
• Build post-training RL infrastructure supporting advanced algorithms and robot interaction.
• Analyze system performance, benchmark, and share findings through code and docs.

Conhecimentos

Python programming

Distributed systems

Debugging skills

Formação

Bachelor's/Master's/PhD in CS/CE/EE or related field

Descrição da vaga

NVIDIA's Cosmos Lab Infrastructure team is seeking an intern to help develop training and post-training systems for Physical AI models, including world foundation models and robot policies. You’ll work with a mentor on a focused project and implement improvements on real AI workloads using NVIDIA's GPU infrastructure.

You will contribute to training, inference, simulation, and evaluation pipelines, collaborating with researchers to optimize performance and scalability.