Senior Python Compiler Engineer

Details

We are looking for experienced software engineers to join our team bringing delightful developer and user experiences to the Python compiler ecosystem. Our goal is to grow NVIDIA’s accelerated Python offerings to make Python one of the first-class citizens for programming NVIDIA CUDA GPUs.

Responsibilities

As a member of our team, you will use your design abilities, coding expertise, creativity, and community engagement to enhance the functionality and performance of NVIDIA GPUs such that Python users can take full advantage of the NVIDIA CUDA platform, including both NVIDIA hardware and software. Specifically, you will be working to:

• Architect, prioritize, and develop new features in CUDA Python
• Analyze, identify, and improve the UX and performance of CUDA software in Python
• Write effective, maintainable, and well-tested code for production use
• Address unique challenges in developing and deploying Python GPU solutions
• Engage with the Python community to develop and drive necessary protocols and standards for the NVIDIA CUDA platform
• Evangelize CUDA programming in Python to encourage and empower adoption of the NVIDIA CUDA platform

Requirements

• BS, MS or PhD degree in Computer Science, Electrical Engineering or related field (or equivalent experience)
• 6+ years experience in compiler development and code optimization
• Strong Python programming skills with a track record of driving formulation or adoption of community standards
• Fluent C, C++, and CUDA C++ programming skills
• Experience designing, developing, tuning, navigating, and/or maintaining a large, complex, multi-language software stack (between C, C++, CUDA C++, and Python)
• Good written communication, collaboration, and presentation skills with the ability of working across team boundaries
• Knowledge of Numba, NumPy, SciPy or a similar framework

Ways To Stand Out From The Crowd

• Deep understanding of the CUDA programming model and language features
• Familiarity with Python ecosystem, language idioms, and pioneering solutions
• Dexterity with compilers, static and dynamic analysis techniques, and dynamic code generation
• Experience developing with the LLVM and MLIR compiler infrastructure
• Experience in memory management of a multi-language project or development of domain-specific libraries/languages for AI, Data Analytics, or Scientific Computing

NVIDIA's invention of the GPU revolutionized parallel computing. 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. Our GPUs are being used in many of the largest high-performance computing projects around the world, solving real-world problems. Our products are used to build and parallelize the most meaningful scientific applications for weather modeling, climatology, fluid dynamics, and defense. We support real science and scientists throughout the world.

With competitive salaries and a generous benefits package, NVIDIA is widely considered to be one of high technology's most desirable employers. We have some of the most forward-thinking and experienced people in the world working for us. Our goal is to craft an environment where we can do our life's best work. If you're creative, autonomous, and highly motivated, we want to hear from you!

The base salary range is 180,000 USD - 339,250 USD. Your base salary will be determined based on your location, experience, and the pay of employees in similar positions. You will also be eligible for equity and benefits.