Join NVIDIA’s Cosmos team as an experienced engineer focused on building agentic systems that enhance the ML lifecycle through automation and intelligence. In this role, you will design and implement workflows for data generation, evaluation, debugging, training orchestration, and iteration, leveraging Python and PyTorch to create robust, modular software. You’ll build AI-native systems where models and agents interact with codebases and experiments to improve productivity, creating self-improving loops that drive better decisions across the system. Key responsibilities include scaling evaluation platforms, integrating open-source components into unified systems for rapid experimentation, and maintaining high standards of engineering excellence through testing, reproducibility, and maintainability. Ideal candidates have extensive experience in ML software platform development, deep Python expertise, and a track record of building impactful developer tooling or workflow automation at scale.
Skills
Python
PyTorch
ML pipelines
Evaluation platforms
Developer tooling
Workflow automation
System design
Testing
Packaging
Debugging
LLM-based systems
Kubernetes
CI/CD
PostgreSQL
Git
Docker
AWS
GCP
What you'll do
Design and implement agentic workflows across the ML lifecycle.
Build AI-native systems enabling models and agents to interact with codebases and tools.
Create self-improving loops for data generation, failure detection, and output evaluation.
Own and evolve large-scale Python and PyTorch codebases for robust software development.
Design and scale evaluation platforms combining automated metrics and human feedback.
Build multimodal ML pipelines covering data processing, experimentation, benchmarking, and deployment.
What we're looking for
Significant experience building machine learning systems and software platforms.
Expert-level Python skills with strong judgment on code modularity and long-term health.
Deep familiarity with PyTorch for debugging, adapting, and extending model behavior.
Experience in building ML pipelines, evaluation systems, and developer tooling at scale.
Strong software engineering fundamentals including testing, packaging, and collaborative practices.
Background in building agent-based systems that perform real-world tasks efficiently.
BS, MS, or equivalent experience in Computer Science or related field with 12+ years of relevant development.
Market check
Salary context
This
$224,000–$356,500
range sits above
93%
of similar postings on FindRole.
Peer median band
$169,500–$249,600
Median floor and ceiling across peers.
Typical midpoint (25–75%)
$162,000–$247,562
Middle half of comparable postings.
Based on 240 comparable postings.*
* 240 is the maximum number of comparable postings sampled.
Employer
About Nvidia
Nvidia is a leading designer of graphics processing units (GPUs) and system-on-chip units, powering gaming, professional visualization, data centers, and artificial intelligence workloads. Industry: Semiconductors & AI Computing
Nvidia currently has
801 open roles
on FindRole.
Listed pay typically runs
$184,000–$287,500
across 797 roles with salary data.