[Remote] Senior AI Compiler Engineer - Applied Research

Note: The job is a remote job and is open to candidates in USA. NVIDIA is a leading technology company specializing in AI infrastructure and GPUs. They are seeking a Senior AI Compiler Engineer to develop innovative technologies in machine learning compilers and AI systems, focusing on low-level optimization and compiler engineering.ResponsibilitiesDesign and implement AI-based technology addressing core problems of low-level GPU code generationBuild SFT and RL training pipelinesDefine model inputs using low-level compiler representationsDefine, implement, and evaluate strategies for intelligent prompt engineering in compilation domainPrototype and iterate on model architectures, prompts, and training strategies for NP-hard problems in optimizing compilersPrepare datasets from compiler traces, optimization passes, and target-specific performance signalsApply RL techniques to optimize for downstream objectives and run rigorous experiments, analysis, and benchmarking across workloads and hardware targetsBuild rigorous benchmarks to assess code quality, correctness, and generation overheadPartner with compiler engineers to integrate and ship learned policies with production toolchainsSkillsM.S. or PhD degree in Computer Engineering, Computer Science related technical field (or equivalent experience)5+ years of experience building AI/ML systemsSolid understanding of machine learning fundamentals and experimentation best practicesStrong software engineering skills in Python and C++Hands-on experience training/fine-tuning/post-training large modelsExperience with reinforcement learningReward modeling from non-differentiable signals (binary runtime/compile success, performance counters)Knowledge of prompt-engineering techniques (CoT, chaining/orchestration, context adaptation, etc)Ability to work across research and engineering, from prototype to productionCUDA programming experience and GPU performance familiarityDistributed training/inference at scale (Megatron, NeMo, vLLM, Triton)Experience working with the NVIDIA training stacksFundamentals of construction of optimizing compilersUnderstanding of GPU performance, experience with benchmarking suites and performance profiling toolsKnowledge of formal methods or static analysis for correctness guaranteesBenefitsYou will also be eligible for equity and [benefits](https://www.nvidia.com/en-us/benefits/).Company OverviewNVIDIA is a computing platform company operating at the intersection of graphics, HPC, and AI. It was founded in 1993, and is headquartered in Santa Clara, California, USA, with a workforce of 10001+ employees. Its website is https://www.nvidia.com.Company H1B SponsorshipNVIDIA has a track record of offering H1B sponsorships, with 448 in 2026, 1872 in 2025, 1354 in 2024, 976 in 2023, 835 in 2022, 601 in 2021, 529 in 2020. Please note that this does not guarantee sponsorship for this specific role.