Be an early applicantLess than 25 applicants

Company

NVIDIA · 7 hours ago

Deep Learning Compiler Engineer

Pennsylvania, United States

Full-time

Remote

Senior Level

$180K/yr - $339K/yr

6+ years exp

Maximize your interview chances

Artificial Intelligence (AI)GPU

Growth Opportunities

H1B Sponsor Likely

Insider Connection @NVIDIA

Discover valuable connections within the company who might provide insights and potential referrals.
Get 3x more responses when you reach out via email instead of LinkedIn.

Responsibilities

Innovate and develop new machine learning compiler and systems technologies

Design, implement, and optimize compilers for high impact AI workloads

Co-design learning system solutions with current and future ML compiler and algorithm technologies.

Collaborate closely with other engineering teams at NVIDIA to build high impact solutions for machine learning acceleration

Qualification

Find out how your skills align with this job's requirements. If anything seems off, you can easily click on the tags to select or unselect skills to reflect your actual expertise.

Machine Learning CompilersDeep Learning FrameworksPython ProgrammingC/C++ ProgrammingAI FrameworksMachine CompilersGPU Performance OptimizationMachine Learning Systems ResearchOpen Source Contributions

Required

Bachelor's degree in Computer Science, Electrical Engineering, or related field (or equivalent experience)

6+ years (academic/ industry) experience with ML/DL systems development preferable for compilers

Strong experience in developing or using deep learning frameworks (e.g. PyTorch, JAX, TensorFlow, ONNX etc.)

Strong python and C/C++ programming skills

Preferred

MS or PhD are preferred

Expertise in AI frameworks such as PyTorch, TensorFlow, and ONNX

Expertise in machine compilers (e.g. Apache TVM, MLIR)

Strong experience in GPU performance optimizations

Strong experience machine learning systems research and productionization

Open source project ownership or contributions

Benefits

Equity

Benefits

Company

NVIDIA

Glassdoor

4.6

NVIDIA is a computing platform company operating at the intersection of graphics, HPC, and AI.

Founded in 1993

Santa Clara, California, USA

10001+ employees

https://www.nvidia.com

H1B Sponsorship

NVIDIA has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)

Distribution of Different Job Fields Receiving Sponsorship

Represents job field similar to this job

Trends of Total Sponsorships

2023 (735)

2022 (892)

2021 (696)

2020 (534)