Senior ML Software Engineer

NVIDIA's Cosmos team is seeking a Senior ML Software Engineer to help build the foundation for the next generation of Physical AI. This role focuses on developing robust, production-grade open-source ML software to support machine learning and gen-AI research at scale. The position is perfect for expert Python engineers passionate about building production-ready systems and making lasting impact through open-source contributions.

As part of NVIDIA, the world leader in accelerated computing, you'll work on cutting-edge ML infrastructure projects, designing and optimizing pipelines for groundbreaking ML models. The role involves significant collaboration with research and product teams, contributing to open-source projects (https://github.com/nvidia-cosmos), and building tools for scalable AI experimentation.

The ideal candidate brings expert-level Python proficiency, strong experience with ML frameworks like PyTorch, and deep understanding of ML system design. You'll need 5+ years of relevant experience and should be comfortable with containerization, CI/CD, and production environments. Standing out candidates will have open-source contributions, experience with GPU computing, and familiarity with distributed computing tools.

The position offers competitive compensation ($148,000-$356,500 base salary) plus equity and benefits. Working at NVIDIA means joining a diverse, inclusive environment at the forefront of AI and digital twins technology, where your work will directly impact the world's largest industries and society at large.