ML and Agentic Systems Engineer

NVIDIA Gruppe; Santa Clara (CA); On-site; USD 224,000 - 356,500; Full time

Benefits offered by this job

Equity

Benefits

Job summary

NVIDIA Gruppe is seeking a Senior Engineer in Santa Clara, CA, to join the Cosmos team. This role focuses on creating AI-native systems that enhance the efficiency of machine learning workflows. Candidates should have extensive Python and PyTorch experience, along with a solid background in building machine learning systems.

The compensation package includes a base salary range of $224,000 to $356,500 for Level 5 positions, and $272,000 to $431,250 for Level 6 positions, plus equity and comprehensive benefits.

Qualifications

• Significant experience building machine learning systems and software platforms.
• Expert-level Python skills with a focus on modularity and long-term code health.
• Deep familiarity with PyTorch for debugging and extending model behavior.

Responsibilities

• Design and implement agentic workflows across the ML lifecycle.
• Build AI-native systems for interaction with models and agents.
• Integrate components into unified systems for rapid experimentation.

Skills

Python

Machine learning systems

PyTorch

Software engineering fundamentals

LLM-based systems

Education

BS, MS, or equivalent experience in Computer Science, Engineering or related field

Job description

At NVIDIA, we’re not just building the future, we’re generating it! Our Cosmos team is pushing the boundaries of multimodal AI, simulation, and world models. As we enter the next phase, we are building agentic systems that can reason about, build, evaluate, and improve AI systems themselves.

We are building systems where AI doesn’t just run models but helps build them. This role is about creating the meta‑layer of modern ML: the agents, tooling, pipelines, and feedback loops that make model development faster, smarter, and increasingly automated. Rather than focusing on inventing individual model architectures, you will build the systems that help models and teams improve continuously. We are looking for exceptional engineers who are passionate about the idea of AI‑native software engineering: systems where agents can work with code, data, experiments, and evaluations to accelerate how machine learning gets done.

What you’ll be doing:

• Design and implement agentic workflows across the ML lifecycle, including data generation and curation, evaluation, debugging, training orchestration, and iteration.
• Build AI‑native systems in which models and agents can interact with codebases, tools, experiments, and environments to improve developer and researcher productivity.
• Create self‑improving loops where agents help generate data, surface failures, evaluate outputs, and drive better decisions across the system.
• Own and evolve large‑scale Python and PyTorch codebases, turning fast‑moving ideas into robust, modular, reusable software.
• Design and scale evaluation platforms that combine automated metrics, human feedback, and agent‑driven analysis.
• Build and maintain multimodal ML pipelines spanning data processing, experimentation, benchmarking, and deployment.
• Integrate open‑source and internal components into unified systems that enable rapid experimentation and reliable iteration.
• Raise the bar on engineering excellence across the team through strong practices in testing, reproducibility, packaging, code health, and maintainability.

What we need to see:

• Significant experience building machine learning systems and software platforms, not only models.
• Expert‑level Python skills, with strong judgment around modularity, abstraction boundaries, and long‑term code health.
• Deep familiarity with PyTorch, including the ability to debug, adapt, and extend model behavior within larger software systems.
• Experience building pipelines, evaluation systems, developer tooling, or workflow automation for ML at meaningful scale.
• Strong software engineering fundamentals, including system design, testing, packaging, debugging, and collaborative codebase evolution.
• Strong agency in LLM‑based systems, such as tool use, planning, multi‑step workflows, code agents, or automation over data and experiments.
• Comfort operating in fast‑moving environments where ambiguous ideas must be turned into useful systems quickly.
• BS, MS, or equivalent experience in Computer Science, Engineering, or a related field.
• 12+ years of relevant software development experience.

Ways to stand out from the crowd:

• You have built agent‑based systems that do real work: coding, evaluation, data generation, triage, experimentation, or orchestration.
• You have contributed to impactful open‑source ML, Python, or developer tooling.
• Background with context compression and agent memory techniques.
• Familiarity with agent safety and agent identity (AuthN, AuthZ, IAM).
• You bring a high bar for software craftsmanship, but know how to apply it in research‑adjacent environments without slowing innovation down.

Your base salary will be determined based on your location, experience, and the pay of employees in similar positions. The base salary range is 224,000 USD – 356,500 USD for Level 5, and 272,000 USD – 431,250 USD for Level 6.

You will also be eligible for equity and benefits.

NVIDIA uses AI tools in its recruiting processes.

NVIDIA is committed to fostering a diverse 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, color, national origin, gender, gender expression, sexual orientation, age, marital status, veteran status, disability status or any other characteristic protected by law.