Senior Machine Learning Applications and Compiler Engineer, LPX

What you'll do

• Build and maintain high-performance runtime and compiler components for end-to-end inference optimization.
• Map inference workloads to NVIDIA systems; benchmark and profile compiler performance and efficiency.
• Prototype compilation and runtime techniques, collaborate with hardware teams, and publish technical work.

What they ask for

• MS or PhD in Computer Science, Electrical/Computer Engineering, or a related field, or equivalent experience.
• Proficiency in systems programming (C/C++ and/or Rust) and fundamentals including data structures, algorithms, and concurrency.
• Hands-on compiler or runtime development experience, including IR design, optimization passes, or code generation; experience with LLVM and/or MLIR.
• Familiarity with TensorFlow and PyTorch, experience with ONNX, and understanding of parallel and heterogeneous compute architectures.

Nice to have

• Work on spatial or dataflow architectures, or contributions to open-source ML frameworks, compilers, or runtime systems.
• Research impact through relevant publications or presentations; experience with large-scale distributed AI inference or training systems.

Pay

152,000 USD - 241,500 USD (base salary, Level 3); 184,000 USD - 287,500 USD (base salary, Level 4)

Tools

C/C++, Rust, LLVM, MLIR, TensorFlow, PyTorch, ONNX

AI focus

Optimizes compilation and runtime execution for large-scale neural network inference workloads, including graph transformations, scheduling, and memory/layout optimization.