Compiler Engineer – MLIR, PyTorch Infrastructure

🕒 Março 19

🏄 California – Remoto

🏄 California – Remote

⏰ Tempo Integral

🟡 Pleno

🟠 Sênior

👷🏻‍♀️ Engenheiro

🦅 Patrocina Visto H1B

Esta empresa já patrocinou vistos H1B no passado.

Candidatar-se

Encontrar Vagas Remotas Similares

Receber Emails de Vagas Remotas

📊 Verifique sua pontuação de currículo para esta vaga

Melhore suas chances de conseguir uma entrevista verificando sua pontuação de currículo antes de se candidatar.

Enviar Currículo

Mythic

SiteLinkedInTodas as Vagas

11 - 50 funcionários

Fundada em 2012

🤖 Inteligência Artificial

🔧 Hardware

💰 $70.000.000 Series C em 2021-05

Artificial Intelligence • Hardware • Robotics

A Mythic é a empresa líder em computação analógica de alto desempenho, focada em fornecer poder de computação de IA de ponta, eficiente em termos de custo e energia, em vários setores, incluindo robótica, defesa e dispositivos de consumo. A inovadora tecnologia de computação na memória da empresa resolve as limitações da computação digital tradicional ao armazenar parâmetros de IA diretamente no processador, aprimorando significativamente as aplicações de inferência de IA. A Mythic visa capacitar as indústrias com avanços em escalabilidade de desempenho e eficiência, proporcionando uma plataforma robusta para a próxima geração de aplicações de IA.

Descrição

• Architect the migration of the existing compiler flow into MLIR, defining dialects, passes, and lowering strategies. • Build conversion paths between MLIR and Mythic’s custom low-level IR to keep both flows operational during migration. • Define validation infrastructure within MLIR, including interpretation or execution paths for simulation and debugging. • Enable compilation by extending MLIR integration across analog accelerators and digital subsystems. • Leverage Torch-MLIR where PyTorch inputs are available, and guide future integration with PyTorch 2.0 compiler technologies (TorchInductor, TorchDynamo, Torch-MLIR)

🎯 Requisitos

• 3+ years of experience in compiler or high-performance systems development. • Proficiency in modern C++ (C++14/17/20) and Python. • Direct, hands-on experience with MLIR, including dialect design, compiler passes, or lowering pipelines. • Strong understanding of compiler IRs and transformations, with the ability to reason about lowering from high-level ops to hardware-aware representations.

🏖️ Benefícios

• Shape the MLIR and PyTorch integration strategy for novel AI hardware. • Opportunity to contribute upstream and influence the broader ML compiler ecosystem. • Collaborate with a team spanning hardware, runtime, and compiler design. • Competitive compensation, equity, and comprehensive benefits.