NVIDIA is seeking a Senior Machine Learning Applications and Compiler Engineer to develop algorithms and optimizations for their LPX inference and compiler stack. This role involves working at the intersection of large-scale systems, compilers, and deep learning, focusing on how neural network workloads are mapped onto NVIDIA platforms. The engineer will collaborate closely with hardware architects and design teams to influence future architectures and improve performance. This position offers an opportunity to be part of innovative projects in the field of machine learning and compiler development.

• 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).
• 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.
• Solid understanding of parallel and heterogeneous compute architectures, such as GPUs or spatial accelerators.
• Strong analytical and debugging skills, with experience using profiling, tracing, and benchmarking tools.
• Excellent communication and collaboration skills.

• Build, develop, and maintain high-performance runtime and compiler components, focusing on end-to-end inference optimization.
• Define and implement mappings of large-scale inference workloads onto NVIDIA’s systems.
• Extend and integrate with NVIDIA’s software ecosystem, contributing to libraries, tooling, and interfaces.
• Benchmark, profile, and monitor key performance and efficiency metrics to ensure efficient mappings of neural network graphs.
• Collaborate closely with hardware architects and design teams to provide feedback and influence future architectures.
• Prototype and evaluate new compilation and runtime techniques tailored to spatial processors.
• Publish and present technical work on novel compilation approaches for inference and related spatial accelerators.

• Employees at NVIDIA are often offered comprehensive, day-one benefits—including medical, dental, and vision coverage with HSA support, life and disability insurance, an Employee Assistance Program, and a 401(k) with auto-enrollment. Many roles also have generous time off and holidays, donation matching (up to $10,000), and a wide menu of extras like FSAs, commuter benefits, legal and identity-theft protection, pet insurance, and wellness discounts. Optional programs can include student-loan and home-purchase support, plus family care resources and expert medical services.