Назад
Company hidden
13 часов назад

ML Compiler Engineer (AI)

250 000 - 315 000$
Формат работы
onsite
Тип работы
fulltime
Грейд
senior
Английский
b2
Страна
US
Вакансия из списка Hirify.GlobalВакансия из Hirify Global, списка международных tech-компаний
Для мэтча и отклика нужен Plus

Мэтч & Сопровод

Для мэтча с этой вакансией нужен 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

hirify.global 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, прислать код/пароль, запустить код/ПО, не делайте этого - это мошенники. Обязательно жмите "Пожаловаться" или пишите в поддержку. Подробнее в гайде →