Machine Learning Applications and Compiler Engineer, LPX

This position is posted by Jobgether on behalf of a partner company. We are currently looking for a Machine Learning Applications and Compiler Engineer, LPX in Canada.

This role sits at the intersection of machine learning systems, compiler design, and high-performance computing, contributing directly to next-generation AI inference technologies. You will help shape how large-scale neural network workloads are compiled, optimized, and executed across advanced computing architectures. Working in a highly technical and collaborative environment, you will build and refine compiler and runtime systems that directly impact AI performance and efficiency. The role offers exposure to cutting-edge GPU and spatial accelerator technologies, as well as deep collaboration with hardware and software engineering teams. You will contribute to performance-critical infrastructure used to deploy modern AI models at scale. This is an opportunity to work on foundational systems that power advanced machine learning applications globally.

• Develop and maintain high-performance compiler and runtime components focused on AI inference optimization and execution efficiency.
• Design and implement mappings of large-scale neural network workloads onto advanced computing architectures and accelerator systems.
• Build and extend compiler infrastructure, including optimization passes, intermediate representations, graph transformations, and code generation pipelines.
• Profile, benchmark, and analyze system performance to identify bottlenecks and improve execution efficiency across ML workloads.
• Collaborate with hardware architects and cross-functional engineering teams to influence system design and optimize hardware–software co-design.
• Prototype and evaluate new compiler and runtime techniques for scheduling, memory optimization, and graph execution strategies.
• Contribute to tooling, libraries, and frameworks that enable scalable deployment of machine learning models.

Requirements:

• Strong academic background (MS/PhD or equivalent experience) in Computer Science, Electrical/Computer Engineering, or related technical fields.
• Solid software engineering foundation in systems programming languages such as C++ and/or Rust, with strong CS fundamentals in algorithms and concurrency.
• Hands-on experience in compiler or runtime development, including IR design, optimization passes, or code generation.
• Experience working with compiler frameworks such as LLVM and/or MLIR, including custom extensions or passes.
• Familiarity with machine learning frameworks such as TensorFlow or PyTorch and graph formats such as ONNX.
• Understanding of parallel computing, GPU architectures, or other heterogeneous and spatial compute systems.
• Strong analytical and debugging skills, with experience in profiling and performance optimization tools.
• Excellent communication and collaboration skills across multidisciplinary engineering teams.

Benefits:

• Competitive base salary aligned with experience and role level (approx. CAD 105,000–185,000 range depending on level).
• Equity participation and long-term incentive opportunities.
• Comprehensive benefits package including health coverage and employee support programs.
• Opportunity to work on cutting-edge AI infrastructure and next-generation computing systems.
• Exposure to world-class engineering teams and advanced research-driven projects.
• Inclusive and innovation-driven engineering culture focused on technical excellence.
• Career development opportunities in high-performance computing and machine learning systems.