Senior Machine Learning Applications and Compiler Engineer, LPX

NVIDIA, Canada, Toronto +1

The role

NVIDIA is seeking engineers to develop algorithms and optimizations for our LPX inference and compiler stack. You will work at the intersection of large-scale systems, compilers, and deep learning, crafting how neural network workloads map onto future NVIDIA platforms. This is your chance to be part of something outstandingly innovative!

What you need

MS or PhD in Computer Science, Electrical/Computer Engineering, or related field, or equivalent experience, with 5 years of relevant experience.
Strong software engineering background with proficiency in systems level programming (e.g., C/C++ and/or Rust) and solid CS fundamentals in data structures, algorithms, and concurrency.
Hands on experience with compiler or runtime development, including IR design, optimization passes, or code generation.
Experience with LLVM and/or MLIR, including building custom passes, dialects, or integrations.
Familiarity with deep learning frameworks such as TensorFlow and PyTorch, and experience working with portable graph formats such as ONNX.
Solid understanding of parallel and heterogeneous compute architectures, such as GPUs, spatial accelerators, or other domain specific processors.
Strong analytical and debugging skills, with experience using profiling, tracing, and benchmarking tools to drive performance improvements.
Excellent communication and collaboration skills, with the ability to work across hardware, systems, and software teams.
Ideal candidates will have direct experience with MLIR based compilers or other multilevel IR stacks, especially in the context of graph based deep learning workloads.

Ways to stand out from the crowd

Prior work on spatial or dataflow architectures, including static scheduling, pipeline parallelism, or tensor parallelism at scale.
Contributions to opensource ML frameworks, compilers, or runtime systems, particularly in areas related to performance or scalability.
Demonstrated research impact, such as publications or presentations at conferences like PLDI, CGO, ASPLOS, ISCA, MICRO, MLSys, NeurIPS, or similar.
Experience with large-scale AI distributed inference or training systems, including performance modeling and capacity planning for multi rack deployments.

Your base salary will be determined based on your location, experience, and other objective job-related factors. The expected base salary ranges for this position are: 135,000 CAD – 185,000 CAD for Level 3; and 170,000 CAD – 220,000 CAD for Level 4.

You will also be eligible for equity and benefits.

Applications for this job will be accepted at least until November 3, 2026.

This posting is for an existing vacancy.

NVIDIA uses AI tools in its recruiting processes.

Senior Machine Learning Applications and Compiler Engineer, LPX

The role

Senior Machine Learning Applications and Compiler Engineer, LPX

The role

What you will do

What you need

Ways to stand out from the crowd

About NVIDIA

Similar jobs at other companies

Senior Compiler Engineer

Inference Performance Engineer, Agent Driven Inference Optimization

Senior/Staff Deep Learning Compiler Engineer

Sr. Machine Learning - Compiler Engineer III, AWS Neuron, Annapurna Labs

Sr. Machine Learning - Compiler Engineer III, AWS Neuron, Annapurna Labs

Software Engineer, Systems ML - Compilers / Backend

Interview questions

NVIDIA