AI Compiler Formal Verification Intern
At a glance
Junior ML Platform & Ops role at NVIDIA. Switzerland, Zurich · full-time.
Pay not stated ML Platform & Ops salary
aimlroles summary, based on the employer's posting.
What you'll do
- Develop formal verification methods for AI compilers and their features, passes, intermediate representations and lowerings
- Apply semantic verification or static analysis to check compiler behavior for human and agentic use cases
- Explore AI and formal methods to build new ways of verifying AI compilers
What you bring
- You are pursuing a PhD in computer science, computer or electrical engineering, mathematics, or an equivalent program
- You bring strong programming ability in Python or C++
- You have a strong academic foundation in formal methods
Who this fits
This role suits a PhD student focused on formal methods and compiler verification. Experience with LEAN or SMT on compiler-related problems, or familiarity with deep learning frameworks and compilers, can strengthen your fit.
From the employer
NVIDIA's invention of the GPU 1999 sparked the growth of the PC gaming market, redefined modern computer graphics, and revolutionized parallel computing. More recently, GPU deep learning ignited modern AI — the next era of computing — with the GPU acting as the brain of computers, robots, and self-driving cars that can perceive and understand the world. Today, we are increasingly known as “the AI computing company”.
What you’ll be doing:
Research and implement innovative formal methods to verify state of the art AI compilers. This includes but is not limited to AI compiler features, passes, IR's, sub-graphs, lowerings and other critical elements of AI compilers. This role provides you with an opportunity to explore and use AI, formal verification, and other forms of semantic verification or static analysis techniques to innovate solutions that verify AI compilers for human and agentic-use cases.
What we need to see:
• Pursuing PhD in Computer Science, Computer/Electrical Engineering, Mathematics or equivalent program
• Strong Python or C++ programming skills
• Strong academic background in formal methods (preferably PhD)
Ways to stand out from the crowd:
• Academic or industrial background using LEAN or SMT to solve compiler or compiler-adjacent problems
• Knowledge of deep learning frameworks (such as Pytorch and TensorRT) and compilers
With competitive salaries and a generous benefits package, we are widely considered to be one of the technology world’s most desirable employers. We have some of the most forward-thinking and hardworking people in the world working for us and, due to unprecedented growth, our exclusive engineering teams are rapidly growing. If you're a creative and autonomous engineer with a real passion for technology, we want to hear from you.