Skip to content
NEXTMOVEFDE careers · United States
NVIDIA

Senior Software Engineer, Cosmos Infrastructure and End to End Performance

AI summary of the role

Senior software engineer role on the NVIDIA Cosmos team, building world foundation models and the infrastructure for physical AI.

What you’ll do

  • Build state-of-the-art world foundation models (like Cosmos3).
  • Drive end-to-end performance analysis and HW-SW codesign for data center and edge deployments.
  • Engage with customers to ensure Cosmos models are easy to use and enable the ecosystem.
  • Develop infrastructure to automate data ingestion, curation, pre-training, post-training, export/quantization, and edge deployment.

What you’ll bring

  • Master's in Computer Engineering, CS, EE, or related STEM (or equivalent) plus 5 years of relevant experience.
  • Expertise in large-scale parallel and distributed accelerator-based systems.
  • Expertise optimizing performance and AI workloads on large-scale systems, including performance modeling and benchmarking at scale.
  • Proficiency in Distributed PyTorch, Python, C/C++, and strong background in computer architecture, networking, storage, accelerators.

Technologies

PyTorch · Python · C++ · CUDA · TensorFlow · JAX · Megatron-LM · TensorRT-LLM · vLLM · GCP · AWS · Azure

About NVIDIA

Designs and manufactures GPUs and system-on-chips powering data centers, AI workloads, gaming, autonomous vehicles, and HPC. The foundational hardware for modern deep learning.

Public

Source and classification

Internal deployment & tooling · Evidence for this classification:

We're now looking for a Senior Software Engineer, Cosmos Infrastructure and End to End Performance! NVIDIA Cosmos is an open omni-model platform of generative world foundation models (WFMs) designed to accelerate physical AI. By combining world generation, physical reasoning, and action generation into unified systems, Cosmos helps developers simulate physical environments and train robots, autonomous vehicles, and smart spaces. You can explore the project via the NVIDIA Cosmos GitHub or look up releases on Hugging Face. Our team's mission is to build the foundational platform for Physical AI ecosystem enablement, empowering developers to create, train, evaluate, and deploy Physical AI systems through open frontier models and open SOTA training and data curation frameworks. What you will be doing: Building SoTA, World foundation models (https://arxiv.org/pdf/2606.02800) (like
More from the job description

We're now looking for a Senior Software Engineer, Cosmos Infrastructure and End to End Performance! NVIDIA Cosmos is an open omni-model platform of generative world foundation models (WFMs) designed to accelerate physical AI. By combining world generation, physical reasoning, and action generation into unified systems, Cosmos helps developers simulate physical environments and train robots, autonomous vehicles, and smart spaces. You can explore the project via the NVIDIA Cosmos GitHub or look up releases on Hugging Face. Our team's mission is to build the foundational platform for Physical AI ecosystem enablement, empowering developers to create, train, evaluate, and deploy Physical AI systems through open frontier models and open SOTA training and data curation frameworks. What you will be doing: Building SoTA, World foundation models (https://arxiv.org/pdf/2606.02800) (like Cosmos3); Engage with driving end to end performance analysis and drive HW-SW codesign for both the data center infrastructure and Edge deployments Engage with customers to ensure the Cosmos models are easy to use and enabling the ecosystem; Develop infrastructure to improve and automate the entire process of data ingestion, curation, pre-training, post-training, Export/quantization and deployment on the edge; Design for robustness and fault tolerance. What we need to see: A Masters in Computer En [... source excerpt omitted ...] ur location, experience, and the pay of employees in similar positions. The base salary range is 152,000 USD - 241,500 USD for Level 3, and 184,000 USD - 287,500 USD for Level 4. You will also be eligible for equity and benefits. Applications for this job will be accepted at least until September 12, 2026. This posting is for an existing vacancy. NVIDIA uses AI tools in its recruiting processes. NVIDIA is committed to fostering an inclusive work environment and proud to be an equal opportunity employer. As we highly value diversity in our current and future employees, we do not discriminate (including in our hiring and promotion practices) on the basis of race, religion, col

How jobs are selected

Employer postings · Data from · Sources