AI Research Engineer - Applied Scientist Compilers
NVIDIA · Santa Clara, California, US
NVIDIA's GPUs are at the core of modern AI infrastructure, from training large-scale models to running inference in production. That position depends on soft...
Job description
NVIDIA's GPUs are at the core of modern AI infrastructure, from training large-scale models to running inference in production. That position depends on software as much as hardware, and compiler engineering is a big part of what makes it work. We are looking for outstanding AI Research Engineer /Applied Scientist focused on Compilers /Low-level optimization to join the team and develop groundbreaking technologies in machine learning compilers and AI systems. We build innovative AI compiler solutions that work together with NVIDIA's software stack to provide comprehensive acceleration for modern machine learning models. What you'll be doing: Help trailblaze company efforts in applying AI within conventional compilation pipelines. Design and implement AI-based technology addressing core problems of low-level GPU programming. Build training pipelines for supervised fine-tuning and reinforcement learning (RL/RLHF-style or policy optimization variants). Define model inputs/outputs over compiler low level compiler representations. Develop evaluation frameworks to measure code quality, runtime, compile-time overhead, and correctness. Intelligent (domain` task based) prompt engineering. C...