Sign up to access all features of our service.
  • Job search
  • Favorites
  • Create a CV
    New
  • Salaries
  • Subscriptions

Compiler Engineer - MLIR / PyTorch Infrastructure

Latent AI

About us Mythic is building the future of AI computing with breakthrough analog technology that delivers 100× the performance of traditional digital systems at the same power and cost. This unlocks bigger, more capable models and faster, more responsive applications—whether in edge devices like drones, robotics, and sensors, or in cloud and data center environments. Our technology powers everything from large language models and CNNs to advanced signal processing, and is engineered to operate from –40 °C to +125 °C, making it ideal for industrial, automotive, aerospace, and defense. We’ve raised over $100M from world‑class investors including Softbank, Threshold Ventures, Lux Capital, and DCVC, and secured multi‑million‑dollar customer contracts across multiple markets. About the role Join us in advancing the MLIR ecosystem at Mythic. You’ll help extend our existing high‑level dialects and design a new hardware‑aware low‑level dialect, building conversion paths that bridge to our current IRs. Working closely with hardware engineers and ML developers, your work will expand interoperability with PyTorch and other frameworks, laying the groundwork for long‑term innovation. The result: an MLIR‑based stack that unifies hardware constraints with developer needs and accelerates adoption of modern ecosystems. Here’s what you will do 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) Here’s the background we hope you will have 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. The following would be nice to have, but is not required Experience architecting complete MLIR flows: from frontend dialects down to hardware‑aware dialects, including conversion to and from existing IRs. Familiarity with PyTorch compiler technologies, especially Torch-MLIR and integration paths with PyTorch 2.0 (TorchDynamo, TorchInductor). Knowledge of dataflow architectures, scheduling, and memory orchestration. Background in heterogeneous or specialized accelerators (e.g., analog compute, NPUs, GPUs, DSPs). What we offer 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. Team culture At Mythic, we foster a collaborative and respectful environment where people can do their best work. We hire smart, capable individuals, provide the tools and support they need, and trust them to deliver. Our team brings a wide range of experiences and perspectives, which we see as a strength in solving hard problems together. We value professionalism, creativity, and integrity, and strive to make Mythic a place where every employee feels they belong and can contribute meaningfully. #J-18808-Ljbffr Latent AI

Vacancy posted 2 days ago
Similar jobs that could be interesting for youBased on the Compiler Engineer - MLIR / PyTorch Infrastructure in Palo Alto, CA vacancy
  • Latent AI is seeking to enhance the MLIR ecosystem. The ideal candidate will help extend high-level dialects and design a new hardware...  ...Successful applicants should have over 3 years of experience in compiler development, proficiency in modern C++ and Python, and hands-on... 
    Suggested

    Latent AI

    Palo Alto, CA
    2 days ago
  •  ...Principal Compiler Engineer - ML Systems The era of pervasive AI has arrived...  ...innovation in compiler infrastructure and optimization algorithms...  ...anything from digging through PyTorch and machine learning models...  ...Qualifications Experience with MLIR. Familiarity with machine... 
    Suggested
    Full time
    Temporary work
    Local area
    Flexible hours

    SambaNova Systems

    Palo Alto, CA
    1 day ago
  • $80 - $95 per hour

     ...Description Insight Global is looking for a Compiler Engineer with deep expertise in MLIR and LLVM to help extend and evolve our compiler infrastructure for AI/ML workloads with one of our...  ..., Python ML workloads (Tensorflow, PyTorch) Ideally MLIR framework (IREE... 
    Suggested

    Insight Global

    Mountain View, CA
    12 hours ago
  • $200k - $360k

     ...Description About the role Own MLIR dialect design and lowering...  ...tool flows to accelerate compiler development. What we're looking...  ...~5+ years compiler engineering experience, with hands-on MLIR...  ...integrating with ML frameworks (PyTorch, JAX, ONNX, TensorFlow, or equivalent... 
    Suggested
    H1b
    Visa sponsorship
    Work visa

    DensityAI

    Mountain View, CA
    5 days ago
  • $80 - $95 per hour

    A major technology staffing firm is searching for a Compiler Engineer in Mountain View, California. The successful candidate will have expertise in MLIR and LLVM, developing compiler components for AI/ML workloads. Responsibilities include collaborating with hardware teams... 
    Suggested

    Insight Global

    Mountain View, CA
    12 hours ago
  • $140k - $224.25k

     ...the next generation of compiler technologies to...  ...We are looking for an engineer to implement compiler...  ...verification software & related infrastructure in the AI space. You...  ...g., graph rewrites in MLIR, fusion passes, mixed-...  .... This could be PyTorch, JAX/XLA, TensorRT, or... 

    NVIDIA

    Santa Clara, CA
    2 days ago
  • $152k - $241.5k

     ...looking for a Deep Learning Compiler Engineer. NVIDIA is hiring software engineers...  ...and implementing compiler infrastructure techniques for neural...  ...the following technologies: MLIR, XLA, TVM, LLVM, deep learning...  ...frameworks, such as PyTorch. GPU kernel generation... 

    NVIDIA

    Santa Clara, CA
    3 days ago
  • NVIDIA Gruppe is seeking an experienced Compiler Engineer in Santa Clara to design and optimize compiler passes and infrastructure for GPU kernels. You'll work with a dynamic team and be involved in architecture decisions while collaborating across various teams. The ideal... 

    NVIDIA Gruppe

    Santa Clara, CA
    1 day ago
  •  ...feel like one seamless engine. Developers can write...  ...what's possible, who see infrastructure not as a constraint...  ...a Graph Optimization Compiler Engineer to own the middle...  ...working with MLIR, XLA, or similar graph...  ...framework internals — PyTorch eager/compile mode, JAX... 

    Lemurian Labs

    Santa Clara, CA
    2 hours ago
  • $150k - $230k

     ...Stanford researchers and veteran systems engineers who share a vision for redefining the...  ...grow increasingly complex, traditional infrastructure struggles to meet the demands of performance...  ...Work with internals of frameworks like PyTorch, NCCL, CUDA runtime-not as a user, but... 

    Clockwork Inc

    Palo Alto, CA
    3 days ago
  • $176k - $420k

     ...What to Expect As a Software Engineer for the Optimus team, you will build the tools and infrastructure to make and measure improvements to neural network architecture...  ...stable and faster training and validate our PyTorch models What You'll Bring ~ Strong... 
    Hourly pay
    Full time
    Temporary work
    Flexible hours

    Tesla

    Palo Alto, CA
    12 hours ago
  • $184k - $287.5k

     ...We are seeking for an expert Senior Compiler Engineer to join our Compute Compiler Team, with...  ...will work directly with LLVM, Clang, MLIR, and related open‑source projects to upstream...  ...compiler features, refactors, and infrastructure originating from NVIDIA’s downstream... 

    NVIDIA

    Santa Clara, CA
    12 hours ago
  • $170k - $216k

     ...Software Engineer, Test Automation Infrastructure Waymo is an autonomous driving technology company with the mission to be the world's most trusted driver. Since its start as the Google Self-Driving Car Project in 2009, Waymo has focused on building the Waymo Driver... 
    Full time
    Remote work

    Waymo

    Mountain View, CA
    2 days ago
  • $120k - $252k

     ...software development to continuously improve and maintain the test infrastructure used across validation teams within Optimus Hardware...  ...and efficient deployment of new tests and test systems. This engineer supports test engineers on Optimus Validation team. It is the... 
    Hourly pay
    Full time
    Temporary work
    Flexible hours

    Tesla

    Palo Alto, CA
    3 days ago
  • $42 - $60 per hour

     ...testing Experience building automation frameworks and infrastructure to improve efficient of development workflow Experience...  ...particular Possible Teams/Work Focus Areas Database Query Engine (QA), Engineering Systems (ES), Cloud Platform & Deployment... 
    Hourly pay
    Full time
    Casual work
    Internship
    Work at office
    Work from home
    Flexible hours

    Streamlit

    Menlo Park, CA
    9 days ago
  •  ...Opportunity This is a senior, project-oriented IT systems engineering role focused on building scalable, secure, and automated IT...  ...and operational handoff. The Senior IT Systems Engineer - Infrastructure & Automation will partner with the Head of IT, Security... 
    Work at office
    Remote work
    Flexible hours
    2 days per week

    LeoLabs, Inc.

    Menlo Park, CA
    3 days ago
  • $200k - $400k

     ...develop AI-assisted tool flows to accelerate compiler development and performance tuning....  ...register allocation) 5+ years of compiler engineering experience, with hands‑on LLVM target...  ...compilation flow (Optional) Custom‑ISA bring‑up, MLIR experience, GPU / accelerator backends,... 

    DensityAI

    Mountain View, CA
    3 days ago
  • $152k - $241.5k

    NVIDIA in Santa Clara is seeking a Compiler Engineer to drive technical innovation in AI workloads and optimize NVIDIA GPUs. The role involves participating in hands-on development, collaborating across divisions, and solving complex compilation problems. Applicants should... 

    NVIDIA

    Santa Clara, CA
    2 days ago
  • $152k - $241.5k

    NVIDIA Gruppe is looking for skilled software engineers to join the CUDA Tile team, focusing on a new tile-based programming model for...  ...ideal candidates will have at least 3 years of experience in compiler optimization and proficient skills in C/C++ programming. The role... 

    NVIDIA Gruppe

    Santa Clara, CA
    1 day ago
  • $184k - $287.5k

     ...and motivated software engineers to join us and build...  ...optimize GPU kernels and compilers, drive industry...  ...level DSLs and compiler infrastructure to boost kernel developer...  ...ML frameworks (e.g., PyTorch) and inference engines...  ...TorchDynamo/Inductor, MLIR/LLVM, XLA), GPU libraries... 

    NVIDIA

    Santa Clara, CA
    2 days ago
  • $152k - $241.5k

    NVIDIA is hiring an AI & Deep Learning Compiler Engineer for its Deep Learning & AI Compiler (DLC...  ...with compiler technologies (e.g., MLIR, LLVM, XLA, Triton, etc.). Excellent C/...  ...models, algorithms and frameworks, such as PyTorch and JAX. GPU kernel authoring and performance... 

    NVIDIA Gruppe

    Santa Clara, CA
    2 days ago
  • NVIDIA Gruppe is seeking a Senior Deep Learning Software Engineer to develop cutting-edge deep learning solutions in autonomous driving vehicles. You will work on compiler technologies to optimize models for NVIDIA's hardware architecture and collaborate with various teams... 

    NVIDIA Gruppe

    Santa Clara, CA
    2 days ago
  • $152k - $241.5k

     ...looking for versatile software engineers for our XLA team. NVIDIA is...  ...Responsibilities In this role, develop compiler optimization algorithms for...  ...‑source compilers such as MLIR, LLVM and OpenAI Triton....  ...learning frameworks such as JAX, PyTorch or TensorFlow. Extensive... 

    NVIDIA Gruppe

    Santa Clara, CA
    2 days ago
  • Staff Hardware Engineer, CAD Infrastructure Achronix Semiconductor Corporation is a fabless semiconductor corporation based in Santa Clara, California...  ...test CAD tools required for custom layout (e.g. Custom Compiler), Physical Design (e.g. Fusion Compiler), ICV, RedHawk,... 
    Night shift

    Achronix Semiconductor Corporation

    Santa Clara, CA
    2 hours ago
  • $202.5k - $274k

    Intuit Inc. is seeking a Staff Design System Engineer in Mountain View, California. This role will focus on creating foundational UI infrastructure, driving performance optimization, and championing accessibility within the QuickBooks Design System. Ideal candidates will... 

    Intuit Inc.

    Mountain View, CA
    12 hours ago
  • $125k - $300k

    Applied Intuition is seeking a Cloud Infrastructure Engineer in Mountain View, CA. The position focuses on improving the infrastructure for large-scale simulations used in autonomous systems. Ideal candidates will have over 3 years of software engineering experience, particularly... 

    Decisive Point

    Mountain View, CA
    2 days ago
  • $140k - $300k

     ...best coding practices amongst the group, build tools helping engineers to write better code (for instance, performance/memory tracking...  ...Experience with performant software design, object-oriented C++, compiler design and/or hardcore lower-level C code Proficient... 
    Hourly pay
    Full time
    Temporary work
    Flexible hours

    Tesla

    Palo Alto, CA
    1 day ago
  •  ...Software Engineer Matroid is a full-service computer vision company that has developed an end-to-end platform allowing enterprise...  ...a talented Software Engineer to help develop the systems & infrastructure that powers Matroid's computer vision platform. Working alongside... 
    Work experience placement
    Work at office
    Flexible hours

    Matroid

    Palo Alto, CA
    4 days ago
  • $140k - $312k

     ...What to Expect Tesla's continued success depends on Engineers being able to develop, debug, and deploy software quickly. Our web services, tools, and build infrastructure directly impact over 1000 vehicle, energy, and Autopilot Software Engineers. You'll be joining... 
    Hourly pay
    Full time
    Temporary work
    Flexible hours

    Tesla

    Palo Alto, CA
    12 hours ago
  • $152k - $241.5k

     ...is building advanced compiler technologies to accelerate...  ...we are looking for an engineer focused on performance...  ..., and automation infrastructure, ensuring that performance...  ...frameworks such as PyTorch, TensorFlow, JAX, or TensorRT...  ...internals (LLVM, MLIR, CUDA compilation flow... 

    NVIDIA

    Santa Clara, CA
    1 day ago

Do you want to receive more vacancies?

Subscribe and receive similar vacancies to Compiler Engineer - MLIR / PyTorch Infrastructure. Be the first to apply!