The Senior Software Engineer for Generative AI Research at NVIDIA will focus on building scalable infrastructure for training the Cosmos foundation model for physical AI. The role involves optimizing data pipelines and designing workflows that support large-scale AI models for robots and autonomous systems. The Cosmos infrastructure team plays a crucial role in enabling research and production to advance efficiently. This position offers a unique opportunity to contribute to the development of systems that enhance AI's ability to perceive, reason, and act in real-world environments.

• A Masters Degree in Computer Science, Computer Engineering, related STEM Degree, or equivalent experience.
• Strong engineering background in distributed systems, ML infrastructure, or large-scale compute/data platforms with 6 years of relevant work experience.
• Proficiency in Python and at least one systems language such as C++, Go, or Rust.
• Experience with orchestration systems, scheduling, and scalable storage or data pipelines.
• Ability to work across teams, drive technical clarity, and deliver robust solutions in complex environments.
• Comfortable bridging research workflows and production-grade systems.

• Design, build, and operate scalable infrastructure for training Cosmos and supporting large-scale data pipelines.
• Develop high-throughput systems for data processing, retrieval, and workflow orchestration.
• Collaborate across research, optimization, and platform teams to accelerate experiments and deployments.
• Improve system reliability, performance, and observability across distributed compute environments.
• Contribute to long-term infrastructure strategy for training, data management, and large-scale compute efficiency.

• Employees at NVIDIA are often offered comprehensive, day-one benefits—including medical, dental, and vision coverage with HSA support, life and disability insurance, an Employee Assistance Program, and a 401(k) with auto-enrollment. Many roles also have generous time off and holidays, donation matching (up to $10,000), and a wide menu of extras like FSAs, commuter benefits, legal and identity-theft protection, pet insurance, and wellness discounts. Optional programs can include student-loan and home-purchase support, plus family care resources and expert medical services.