Senior AI Compiler Engineer, MLIR

🕒 August 15

🏄 California, Texas, +1 more states – Remote

🏄 California – Remote 🤠 Texas – Remote ☕ Washington – Remote

💵 $152k - $287.5k / year

⏰ Full Time

🟠 Senior

🤖 Artificial Intelligence

🦅 H1B Visa Sponsor

This company has sponsored H1B visas in the past.

👻 Ghost score 5%

What it means: how likely this posting is sitting open without active hiring.How to read it: 60% or higher means we think this is a ghost. Lower scores are less of a worry.How we score it: age, reposts, open roles vs team size, how fast jobs close, and whether pay is listed.Fine print: an estimate from public data, not a judgment of the company. Real jobs can score high, so treat it as a hint, not a fact.

Apply Now

Find Similar Remote Jobs

Set Up Email Alert For New Jobs

📊 Check your resume score for this job

Improve your chances of getting an interview by checking your resume score before you apply.

Upload Resume

NVIDIA

WebsiteLinkedInAll Job Openings

10,000+ employees

Founded 1993

🏥 Healthcare

🏭 Manufacturing

🤖 Artificial Intelligence

Healthcare • Manufacturing • Artificial Intelligence

NVIDIA is a leading technology company specializing in accelerated computing and artificial intelligence. NVIDIA pioneers advancements in graphical processing units (GPUs), cloud computing, data centers, and virtual reality, with a focus on gaming, automotive, healthcare, and robotics industries. The company's innovations, such as NVIDIA Omniverse, transform traditional digital processes by enabling high-fidelity simulations and rendering tasks. Their applications span various industries, from autonomous vehicles using NVIDIA DRIVE to healthcare solutions with NVIDIA Clara, and AI-driven analytics and workflows.

📋 Description

• Develop MLIR-based graph representations and optimizations for future GPU architectures • Partner with framework and hardware teams to enable new model patterns and upcoming GPU architectural features • Define APIs and MLIR dialects • Conduct performance optimizations and analysis • Implement compiler optimizations and kernel generation for neural networks • Contribute to general software engineering work • Define project goals and scope • Lead own development efforts • Mentor early career engineers and interns

🎯 Requirements

• Bachelor's, Master's, or Ph.D. in Computer Science, Computer Engineering, a related field, or equivalent experience • 3+ years of relevant work or research experience in performance analysis and compiler optimizations • Experience with compiler technologies such as MLIR, XLA, and LLVM • Excellent C/C++ and Python programming and software design skills • Experience with debugging, performance analysis, and testing • Ability to work independently, define project goals and scope, and lead own development efforts • Strong interpersonal skills and ability to thrive in a fast-moving, dynamic, product-oriented team • Understanding of deep learning models, algorithms, and frameworks such as PyTorch and JAX • Experience with GPU kernel generation targeting high performance and fast build times • Proficiency in GPU architecture with CUDA or OpenCL programming experience

🏖️ Benefits

• Competitive salaries • Generous benefits package • Equity • Benefits