Be part of a pioneering team at NVIDIA, where you'll directly influence the future of accelerated AI. As a Senior Compiler Engineer, your dynamic and technical role will center around compiler innovation. You will lead the design, implementation, optimization, and maintenance of core compiler technologies that accelerate large-scale deep learning workloads. Collaborate closely with the engineers who developed PyTorch for NVIDIA hardware, pushing new features and advancing framework development. You will conduct performance analysis on workloads across thousands of GPUs, identifying optimization opportunities that will inform the future design of Thunder. Additionally, you'll work within a vibrant ecosystem, partnering with leading compiler, library, and systems teams—including experts from nvFuser, TVM, XLA, and CUDA—to translate cutting-edge research into practical, high-impact solutions for the open-source community.

We seek engineers passionate about building powerful, user-centric tools in a fast-paced, collaborative environment. Key qualifications include:

• A Bachelor's, Master's, or Ph.D. in Computer Science or a related technical field (or equivalent experience).
• 8+ years of relevant work experience.
• Proficiency in Python and experience in developing complex, well-tested software systems.
• Practical experience with deep learning frameworks like PyTorch or JAX, with an understanding of model construction and performance challenges.
• A solid foundation in compiler concepts, including abstract syntax trees (ASTs), intermediate representations (e.g., SSA form), program analysis, and code generation.
• Strong communication and collaboration skills, essential for effective work in a distributed, open-source environment.