3 дня назад
Compiler Optimization Engineer (AI)
Мэтч & Сопровод
Для мэтча с этой вакансией нужен Plus
Описание вакансии
Текст:
TL;DR
Compiler Optimization Engineer (AI): Designing and implementing graph-level optimization passes for a heterogeneous AI compiler with an accent on intermediate representation design, operator fusion, layout transformations, and performance analysis. Focus on optimizing throughput and latency across diverse model architectures and hardware targets, while building reliable interfaces between compiler front ends and code generation backends.
Location: Santa Clara, California, or Toronto, Canada
Company
is building a hardware-agnostic software stack that enables AI workloads to run efficiently across different chips, clouds, and system scales.
What you will do
- Design, develop, and maintain the graph optimization layer of a heterogeneous AI compiler.
- Implement graph transformations such as operator fusion, layout propagation, dead code elimination, constant folding, and algebraic simplification.
- Define and evolve the intermediate representation to support optimization of advancing ML model architectures.
- Analyze performance data and deliver measurable improvements in model throughput and latency.
- Collaborate with compiler front-end and code-generation teams to maintain clear IR interfaces and optimization pipelines.
- Develop testing and validation infrastructure and prototype optimization strategies for new models and hardware targets.
Requirements
- Bachelor’s degree in Computer Science, Computer Engineering, or equivalent practical experience.
- 4+ years of compiler experience focused on intermediate representation design or optimization passes.
- Deep knowledge of graph-level optimization techniques, including fusion, tiling, and layout transformations.
- 4+ years of experience with C/C++.
- Strong written and verbal communication skills, including the ability to create concise technical documentation.
Nice to have
- Master’s or PhD in Computer Science or Computer Engineering.
- Experience with polyhedral models, affine analysis, MLIR, XLA, or similar graph-level IR frameworks.
- Knowledge of hardware memory hierarchies, GPU or accelerator performance, and ML framework internals such as PyTorch, JAX/XLA, or TensorRT.
- Understanding of ML architectures, quantization, sparsity, or other model-level optimization techniques.
- Contributions to open-source compiler or ML infrastructure projects.
Culture & Benefits
- Ownership of a critical compiler layer with direct, measurable impact on model performance.
- Work on graph-level AI infrastructure problems across diverse hardware targets and model architectures.
- Competitive compensation with equity and potential company bonus opportunities.
- Medical, dental, and vision coverage, retirement savings, and supplemental wellness benefits.
Будьте осторожны: если работодатель просит войти в их систему, используя iCloud/Google, прислать код/пароль, запустить код/ПО, не делайте этого - это мошенники. Обязательно жмите "Пожаловаться" или пишите в поддержку. Подробнее в гайде →
Похожие вакансии
4 дня назад
Senior ML Compiler Engineer (AI Accelerators)
4 дня назад
Compiler Engineer (AI)
4 дня назад
Compiler Engineer (AI Infrastructure)
180 000 - 400 000$
4 дня назад
Member of Technical Staff, Hardware, Compiler Engineer (AI)
200 000 - 420 000$
4 дня назад
AI Compiler Engineer (AI)
190 000 - 255 000$
4 дня назад