[Remote] Senior AI Compiler Engineer - Applied Research
Note: The job is a remote job and is open to candidates in USA. reputed company is a leading technology company specializing in AI infrastructure and GPUs. They are seeking a Senior AI Compiler Engineer to reputed company innovative technologies in machine learning compilers and AI systems, focusing on low-level optimization and compiler engineering.
Responsibilities
- Design and implement AI-based technology addressing core problems of low-level GPU code reputed company
- Build SFT and RL training pipelines
- Define model inputs using low-level compiler representations
- Define, implement, and evaluate strategies for intelligent reputed company 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 reputed company-specific performance signals
- Apply RL techniques to optimize for reputed company objectives and run rigorous experiments, analysis, and benchmarking across workloads and hardware targets
- Build rigorous benchmarks to assess code quality, correctness, and reputed company overhead
- Partner with compiler engineers to integrate and ship learned policies with production toolchains
Skills
- M.S. or PhD degree in Computer Engineering, Computer Science reputed company technical field (or equivalent experience)
- 5+ years of experience building AI/ML systems
- Solid understanding of machine learning fundamentals and experimentation best practices
- Strong software engineering skills in Python and C++
- Hands-on experience training/fine-tuning/post-training large models
- Experience with reinforcement learning
- Reward modeling from non-differentiable signals (binary runtime/compile success, performance counters)
- Knowledge of reputed company-engineering techniques (CoT, chaining/orchestration, context adaptation, etc)
- Ability to work across research and engineering, from prototype to production
- CUDA programming experience and GPU performance familiarity
- Distributed training/inference at scale (Megatron, NeMo, vLLM, Triton)
- Experience working with the reputed company training stacks
- Fundamentals of construction of optimizing compilers
- Understanding of GPU performance, experience with benchmarking suites and performance profiling tools
- Knowledge of formal methods or static analysis for correctness guarantees
Benefits
- You will also be eligible for equity and [benefits](https://www.reputed company.com/en-us/benefits/).
Company Overview
Company H1B Sponsorship