13 часов назад
ML Compiler Engineer (AI)
250 000 - 315 000$
Мэтч & Сопровод
Для мэтча с этой вакансией нужен Plus
Описание вакансии
Текст:
TL;DR
ML Compiler Engineer (AI): Developing compiler toolchains and optimization passes for a novel LLM accelerator based on hybrid optical-electronic compute with an accent on compiler backends, graph transformations, and hardware-software co-design. Focus on integrating PyTorch, JAX, and Triton, optimizing computational graphs and memory access patterns, and ensuring compiler correctness and performance.
Location: San Jose, California or Austin, Texas; full-time onsite position
Salary: $250,000–$315,000 base salary plus equity
Company
is an AI hardware startup developing silicon photonics and programmable metasurface technology for energy-efficient, high-performance LLM inference.
What you will do
- Lead development of compiler toolchains for a custom LLM accelerator architecture.
- Design compiler components, including IR dialects, graph transformations, lowering passes, and custom backends.
- Develop optimization strategies that connect software algorithms with hardware implementations.
- Optimize computational graphs and memory access patterns for hybrid optical-electronic hardware.
- Integrate the compiler with PyTorch, JAX, Triton, and custom inference engines.
- Build automated testing and continuous integration infrastructure for compiler correctness and performance.
Requirements
- Master’s degree in Computer Science or a related field.
- 5+ years of compiler development experience.
- Expert-level proficiency in Python and C.
- Experience with hardware compilers, compiler frameworks, and code optimization.
- Understanding of LLM architectures, computer architecture, memory hierarchies, and parallel computing.
- Experience with AI/ML accelerators such as GPUs, TPUs, or FPGAs and their programming models.
Nice to have
- PhD in Computer Science or a related field and 7+ years of industry experience.
- Experience with graph theory, graph rewriting, DAG optimization, MLIR, and AST processing.
- Experience debugging and instrumenting parallel programs.
- Knowledge of LLM quantization, neural network optimization, high-performance computing, or low-latency system design.
- Experience with structured, human-supervised AI or agentic coding workflows.
Culture & Benefits
- Collaborative engineering environment focused on photonics, AI, and hardware-software co-design.
- Health plan premiums covered at 100% for employees and dependents, with HSA contributions.
- Unlimited PTO.
- 401(k) matching and stock option opportunities.
- Dental, vision, life, hospital, critical illness, and accident insurance options.
Будьте осторожны: если работодатель просит войти в их систему, используя iCloud/Google, прислать код/пароль, запустить код/ПО, не делайте этого - это мошенники. Обязательно жмите "Пожаловаться" или пишите в поддержку. Подробнее в гайде →
Похожие вакансии
9 часов назад
Compiler Engineer (AI Infrastructure)
180 000 - 400 000$
14 часов назад
AI Compiler Engineer (AI)
190 000 - 255 000$
13 часов назад
Research Engineer (AI)
170 000 - 300 000$
15 часов назад
AI Research Engineer (LLM)
13 часов назад
AI Engineer
159 000 - 195 000$
13 часов назад
AI Engineer (LLM)
200 000 - 400 000$