Senior AI Compiler Engineer - Applied Research

NVIDIA is a leader in AI infrastructure, and they are seeking a Senior AI Compiler Engineer to develop innovative AI compiler solutions. The role involves designing AI technologies for GPU code generation, building training pipelines, and collaborating with compiler engineers to enhance production toolchains.

Responsibilities:

• Design and implement AI-based technology addressing core problems of low-level GPU code generation
• Build SFT and RL training pipelines
• Define model inputs using low-level compiler representations
• Define, implement, and evaluate strategies for intelligent prompt engineering in compilation domain
• Prototype and iterate on model architectures, prompts, and training strategies for NP-hard problems in optimizing compilers
• Prepare datasets from compiler traces, optimization passes, and target-specific performance signals
• Apply RL techniques to optimize for downstream objectives and run rigorous experiments, analysis, and benchmarking across workloads and hardware targets
• Build rigorous benchmarks to assess code quality, correctness, and generation overhead
• Partner with compiler engineers to integrate and ship learned policies with production toolchains