Senior Deep Learning Compiler Engineer – XLA

🔥 13 hours ago

🏄 California, Texas, +1 more states – Remote

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

💵 $152k - $241.5k / year

⏰ Full Time

🟠 Senior

👷🏻‍♀️ Engineer

🦅 H1B Visa Sponsor

This company has sponsored H1B visas in the past.

👻 Ghost score 1%

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 compiler optimization algorithms for deep learning workloads • Optimize inference and training performance for the JAX framework and the OpenXLA compiler on NVIDIA GPUs at scale • Collaborate with partners in deep learning framework teams and hardware architecture teams • Craft and implement compiler optimization techniques for deep learning network graphs • Design graph partitioning and tensor sharding techniques for distributed training and inference • Perform performance tuning and analysis • Implement code generation for NVIDIA GPU backends using open-source compilers such as MLIR, LLVM and OpenAI Triton • Design user-facing features in JAX and related libraries • Perform general software engineering work • Work closely with GPU hardware engineering teams to design AI compiler software features for next-generation GPUs

🎯 Requirements

• Bachelors, Masters or Ph.D. in Computer Science, Computer Engineering, related field (or equivalent experience) • 4+ years of relevant work or research experience in performance analysis and compiler optimizations • Ability to work independently, define project goals and scope, and lead your own development effort adopting clean software engineering and testing practices • Excellent C/C++ programming and software design skills, including debugging, performance analysis, and test design • Strong foundation in architecture of CPU, GPUs or other high performance hardware accelerators • Knowledge of high-performance computing and distributed programming • CUDA or OpenCL programming experience is desired but not required • Experience with XLA, TVM, MLIR, LLVM, OpenAI Triton, deep learning models and algorithms, and deep learning framework design is a huge plus • Strong interpersonal skills • Ability to work in a dynamic product-oriented team • History of mentoring junior engineers and interns is a bonus • Experience working with deep learning frameworks such as JAX, PyTorch or TensorFlow is a way to stand out • Extensive experience with CUDA or GPUs in general is a way to stand out • Experience with open-source compilers such as XLA, LLVM, MLIR or TVM is a way to stand out

🏖️ Benefits

• Equity • Benefits package • Competitive salary