Senior DFX Software Engineer - Machine Learning

NVIDIA•Us, CA

About The Position

We are looking for a Senior DFX Software Engineer - Machine Learning who enjoys thinking creatively and solving challenges that require innovation. On our team, we define and build methodologies, software, and flows tailored to the field of silicon device testing, silicon debug, and silicon failure analysis. Our success is attributed to our people, who are among the brightest in the world, and a company culture that fosters and encourages innovation and fuels creativity. Our team contributes to the advancement of all fields in which NVIDIA participates, from gaming to building groundbreaking state-of-the-art compute platforms and Artificial Intelligence, by enabling high-quality Silicon defect screening to sustain all these fields. This often requires new ways of thinking to meet new challenges, and we pride ourselves on our ability to tackle these challenges in ways that enable our success. If you are a like-minded person who enjoys innovation and likes to solve technical challenges, then we would love to hear from you.

Requirements

BS in EE or CS (or equivalent experience); MS or higher degree preferred
5+ years of experience in Software development
Strong programming experience in Python or C++
Hands on development in modern C++ is a huge plus
Experience use of LLMs (Large Language Models), GNNs (Graph Neural Networks), and Reinforcement Learning for efficient EDA solution
Expertise in high performance algorithms for DFT, simulations, and failure analysis
Understanding of different agent architectures, RAG systems, and communication protocols
Deep familiarity with reinforcement learning algorithms like PPO, SAC, or Q-learning, including experience tuning hyper-parameters and reward functions
Hands on experience with large scale training (e.g., ZeRO) and data processing (e.g. Spark)
Excellent communication skills

Nice To Haves

Proven deployment of large-scale agentic application with high concurrency and agility
Experience with software and hardware especially involving DFT, failure analysis, and CAD tools
Working experience of agentic models / frameworks, observability and evaluation tools
In-depth understanding of the graph neural networks, and reinforcement learning for logic design automation
Experience with fine-tuning large language models, building advanced multi-agent systems, RAG pipelines and vector databases

Responsibilities

Develop high performance software to enable design and development of efficient test pattern generation, application of these patterns on Silicon, failure analysis, and yield learning
Create efficient parallel graph traversal and graph analysis techniques
Work with multi-functional teams to assess and tackle problems that involve multiple areas of expertise through the company
Apply LLMs, RAGs, graph-based ML approaches, and reinforcement learning to define innovative solutions