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
- 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
- ...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...SuggestedFull timeTemporary workLocal areaFlexible hours
$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$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...SuggestedH1bVisa sponsorshipWork visa$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$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...$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 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...
- ...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...
$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...$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 payFull timeTemporary workFlexible hours$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...$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 timeRemote work$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 payFull timeTemporary workFlexible hours$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 payFull timeCasual workInternshipWork at officeWork from homeFlexible hours- ...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 officeRemote workFlexible hours2 days per week
$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,...$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...$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...$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...$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 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...
$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...- 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
$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...$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...$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 payFull timeTemporary workFlexible hours- ...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 placementWork at officeFlexible hours
$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 payFull timeTemporary workFlexible hours$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...
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to Compiler Engineer - MLIR / PyTorch Infrastructure. Be the first to apply!
- principal infrastructure engineer Palo Alto, CA
- lead infrastructure engineer Palo Alto, CA
- remote infrastructure engineer Palo Alto, CA
- data infrastructure engineer Palo Alto, CA
- senior infrastructure engineer Palo Alto, CA
- infrastructure engineer Palo Alto, CA
- infrastructure automation engineer Palo Alto, CA
- infrastructure developer Palo Alto, CA
- infrastructure engineering manager Palo Alto, CA
- security infrastructure engineer Palo Alto, CA

