Principal AI Compiler Engineer

Principal AI Compiler & Runtime EngineerUS (Hybrid) About the RoleBe part of the team creating the software foundation for next-generation AI compute platforms. In this role, you'll work on compiler technologies, AI runtimes, graph optimization, and hardware-aware execution in close collaboration with inference engineers, ML scientists, and hardware specialists. You'll leverage modern AI technologies and methodologies to accelerate software development, improve software-hardware co-design, and optimize the deployment and execution of AI workloads on next-generation compute platforms.This position offers the opportunity to contribute to state-of-the-art AI infrastructure, optimize software for emerging AI hardware, and help define how modern machine learning workloads are represented, compiled, and executed at scale.We are particularly interested in engineers who have applied AI technologies to solve complex systems, compiler, runtime, or hardware challenges, rather than solely using AI as a software productivity tool. How You'll ContributeBuild and optimize compiler and runtime infrastructure for modern AI workloadsEnable efficient execution of machine learning models across GPUs, NPUs, TPUs, and custom AI acceleratorsApply modern AI technologies and methodologies to improve software development, system optimization, and software-hardware co-design processesCollaborate with inference, systems, and hardware teams to improve software-hardware co-designInvestigate and resolve performance bottlenecks through profiling, benchmarking, and system-level analysis Required SkillsBSc, MSc, or PhD in Computer Science, Engineering, Mathematics, or a related disciplineStrong programming skills in C/C++ and/or Python in Linux environments using common development toolsSolid understanding of machine learning fundamentals and modern AI workloadsExperience applying AI technologies to software engineering, performance optimization, and hardware-aware design challenges Examples of Relevant Backgrounds IncludeAI compiler frameworks and infrastructure (e.g., Mojo/Max, MLIR, LLVM, XLA, OpenXLA, Triton, Gluon)Compiler optimizations such as operator fusion, graph transformations, scheduling, code generation, and lowering pipelinesAI runtimes and execution frameworks (e.g., ONNX Runtime, TensorRT, TVM Runtime, IREE Runtime, XLA)Deep understanding of AI systems, with experience applying AI technologies to solve engineering, performance, systems, or hardware-software optimization challengesPerformance optimization of machine learning workloads on GPUs, TPUs, NPUs, DSPs, or custom acceleratorsHardware-aware software development and AI accelerator enablementDevelopment of high-performance kernels and operators (e.g., GEMMs, convolutions, attention, normalization, quantization)Distributed AI training or inference systemsModel execution frameworks such as Max, PyTorch, TensorFlow, JAX, or ONNX Nice to HaveExperience with Modular (Mojo/Max), OpenXLA, StableHLO, Torch-MLIR, Triton, TVM, or IREEExperience developing software for AI accelerators or machine learning hardware platformsContributions to open-source projects such as LLVM, MLIR, PyTorch, OpenXLA, Triton, Gluon, or xDSLExperience building software for AI accelerators or AI compute platforms Location & TravelHybrid role based in USAbility to travel periodically for collaboration with global teams and stakeholdersInternational travel may be required Compensation & Benefits Our benefits package for this position includes medical, dental and vision coverage; 401(k) with Company match; unlimited vacation, paid holidays; paid leave for personal events; parental benefits and employee assistance program.All compensation and benefits are subject to the terms of the applicable plan documents and Company policies, which may be amended or discontinued at the Company’s discretion. Eligibility for and the value of any bonus or equity award is not guaranteed. Equal Opportunity HTEC Group, Inc. is an equal opportunity employer.