Jobs in the United States

25+ jobs

Save this search

Added 1 min ago
K
Lead Mechanical Engineer
Kiewit
Lone Tree, CO
- $100,000 USD / year
Support development of project schedules, budgets, and resource plans for mechanical engineering scope.
Added 1 min ago
Sales Engineer, VisionLink Full Fleet Solutions
Caterpillar, Inc.
Westminster, CO
Partner with Account Executives, Business Development Managers, and Dealer Sales teams throughout the sales cycle to drive complex fleet management opportunities.
Added 2 mins ago
VH
Biomedical Equipment Support Specialist (HTM)
Veterans Health Administration
Hines, IL
- $84,528 to $109,893 USD / year
…medical equipment. Biomedical support requires the application of engineering and mechanical concepts and methodology to provide lifecycle management and technical support of diagnostic,…
Added 2 mins ago
VH
Medical Instrument Technician (Echocardiography)
Veterans Health Administration
Dayton, OH · Richmond, IN
- $79,091 to $102,824 USD / year
…echocardiograms. The echocardiographer works with biomedical engineering to maintain all cardiac ultrasound equipment. The incumbent ensures timely transfer of diagnostic images to the…
Added 2 mins ago
Digital Innovation Division Mission Solutions Specialist
The Aerospace Corpor
Colorado Springs, CO
…expertise. As the operator of a federally funded research and development center (FFRDC), we are broadly engaged across all aspects of space- delivering innovative solutions that span…
Added 2 mins ago
Technical Infrastructure Program Manager - Long Haul, Fiber Deployment, GCID
Amazon
Denver, CO
Program management skills to monitor vendor delivery of physical infrastructure projects and delivering high quality fiber to make ready for circuit and data center turn-up. Technical…
Added 2 mins ago
R&D Machinist (onsite)
Collins Aerospace
Pueblo, CO
U.S. citizenship is required, as only U.S. citizens are authorized to access information under this program/contract.
Added 4 mins ago
VH
Medical Instrument Technician (Echocardiography)
Veterans Health Administration
Rumford, ME · Chelsea, ME · Lincoln, ME · Bangor, ME · Augusta, ME · Portland, ME
- $80,024 to $104,037 USD / year
…echocardiograms. The echocardiographer works with biomedical engineering to maintain all cardiac ultrasound equipment. The incumbent ensures timely transfer of diagnostic images to the…
Added 7 mins ago
VH
Biomedical Equipment Support Specialist (Information Systems)
Veterans Health Administration
Hines, IL
- $101,313 to $131,710 USD / year
…facility. Works in support of the medical equipment management program. In addition to duties at the full performance level, the BESS maintains medical servers and networks, diagnostic…
Added 18 mins ago
Junior R&D / Pilot Plant Technician
Dow
South Charleston, WV
Dow has an exciting opportunity for a Junior R&D / Pilot Plant Technician supporting Research & Development activities in Freeport TX and South Charleston, WV .
Added 21 mins ago
SS
IT Specialist (Operating Systems Technician)
Social Security Administration
Durham, NC · Woodlawn, MD
- $77,983 to $111,087 USD / year
At the full performance level, Serves as an IT Specialist in the Office of Systems Operations and Hardware Engineering and is responsible for the complex analysis, planning, design,…
Added 26 mins ago
PT
Embedded Software Engineer
PDS Tech Commercial
Indianapolis, IN
- $78.49 USD / hour
Engineer the Connected Future of HVAC as an Embedded Software Engineer
Added 27 mins ago
Software Engineer - E2
Lockheed Martin
Sunnyvale, CA · Huntsville, AL · Littleton, CO · Sunnyvale · Huntsville · Littleton
Plans, conducts, and coordinates software development activities. Designs, develops, documents, tests, and debugs software that contains logical and mathematical solutions to…
Added 27 mins ago
Software Engineer Sr - E3
Lockheed Martin
Boulder, CO · Longmont, CO · Boulder · Longmont
Plans, conducts, and coordinates software development activities. Designs, develops, documents, tests, and debugs software that contains logical and mathematical solutions to…
Added 27 mins ago
Project Engineering Mgr - L5
Lockheed Martin
Littleton, CO · Littleton
In this role, you will act as the primary interface between the Fleet Ballistic Missile Program and Space Production Operations ensuring clear two-way communication in support of our…
Added 27 mins ago
Cyber Systems Security Engineering Prin - E6
Lockheed Martin
Virtual Two, MD · MD
…Excellence. The role will define and govern secure architectures, engineering practices, accreditation strategies, and technical integration across a complex, multi-partner, five-year…
Added 27 mins ago
Software Quality Engineer - E2
Lockheed Martin
Littleton, CO · Littleton
Develops, modifies and applies quality principles to the development and use of software and software-based systems for make and/or buy software products. Performs verification and…
Added 27 mins ago
Systems Engineer Sr - E3
Lockheed Martin
Littleton, CO · Littleton
The Program’s focus is on meeting the current mission needs of our customer, but also pioneering future automation solutions for the NGA based on agency directives. We are developing…
Added 27 mins ago
Systems Engineer-IT Sr - Level 3 (Secret)
Lockheed Martin
Littleton, CO · Littleton
…information technology related responsibilities including design, development, and analysis of computer systems or programs based on and related to user or system design specifications;…
Added 27 mins ago
Systems Engineer Stf - E4
Lockheed Martin
Fort Worth, TX · Fort Worth
…studies to translate customer requirements into hardware and software…
Added 27 mins ago
Missile Systems Integration/Electrical Test Engineer - E2
Lockheed Martin
Cape Canaveral, FL · Cape Canaveral
As a Systems Integration & Test Engineer, you will plan and execute the integration test program for the Eastern Range Sustainment Electrical Test Team. You will develop test documentation…
Added 27 mins ago
Software Engineer Intern - Summer 2027
Lockheed Martin
King of Prussia, PA · King of Prussia
Performs assignments in technical/engineering organizations (e.g., Electronics Engineering, Software Engineering, Facilities Engineering, Manufacturing Engineering) during vacations and…
Added 28 mins ago
Systems Engineering Asc Mgr - L4
Lockheed Martin
Orlando, FL · Orlando
Be a part of the Digital Sustainment Analytics (DSA team at Lockheed Martin. The (DSA) Digital Tools and Engineering (DT&E) Team develops software and data solutions across the company…
Added 28 mins ago
Systems Engineer Sr - E3
Lockheed Martin
Owego, NY · Owego
This role supports the development of collaborative-autonomy-enabled small UAS applications, including Group 1 / 2 UAV development and airborne launch operations. You will work across…
Added 28 mins ago
Software Engineer Sr - E3
Lockheed Martin
Owego, NY · Owego
This role focuses on creating software solutions for complex, collaborative-autonomy-enabled small UAS applications, including Group 1 and 2 UAV development and airborne launch operations.…

NVIDIA USA

Senior Compiler Engineer - AI

Remote, United States

The expected base salary range for this position is 184,000 USD – 287,500 USD.

Apply About employer

Applications open in a new tab.

Job details

Pay: The expected base salary range for this position is 184,000 USD – 287,500 USD.
Experience: 6+ years of relevant work or research experience in compilers, distributed systems, or deep learning training.
Posting language: English
Posted: Nov 2, 2026

Original job description

NVIDIA's invention of the GPU in 1999 sparked the growth of the PC gaming market, redefined modern computer graphics, and revolutionized parallel computing. More recently, GPU deep learning ignited modern AI - the next era of computing - with the GPU acting as the brain of computers, robots, and self-driving cars that can perceive and understand the world. Today, we are increasingly known as "the AI computing company".

NVIDIA is hiring world class Software Engineers with AI Compiler and distributed systems experience for its Deep Learning Compiler Technology team. The team builds JaxPP, NVIDIA's open-source library for pipeline parallelism in JAX. JaxPP turns a JAX training program into per-stage tasks and runs them across many GPUs. With it, very large models can be trained on NVIDIA GPU clusters using unmodified JAX and XLA. Join the team that is developing advanced compiler and runtime technology for the entire deep learning community.

What you'll be doing:

Design and implement program transformations on JAX programs (jaxpr) for large-scale training, on top of unmodified JAX and XLA: pipeline schedules, placement of tasks across stages, sharding (including JAX explicit sharding), and overlap of communication with computation.
Develop technologies and products involving optimizing compilers and runtime execution frameworks for accelerating deep learning training on large scale distributed systems.
Work on compiling a graph intermediate representation (IR) to target specific code, and build the distributed runtime that dispatches per-stage programs to many GPUs and moves data between stages through communication libraries such as NCCL.
Contribute to advancing the state of the art in compiler technology for delivering high performance on large-scale training workloads, including mixture-of-experts models.
Collaborate with members of the software framework teams (for example JAX and XLA) and the hardware architecture teams to accelerate the next generation of deep learning software, and work on integrating JaxPP into JAX training frameworks, including MaxText.
Be a core contributor to the open-source JaxPP project: keeping it working with new JAX releases, testing, performance tracking, and working with teams who adopt it for their training runs.

What we need to see:

Masters or PhD in Computer Science, Computer Engineering, or related field or equivalent experience.
6+ years of relevant work or research experience in compilers, distributed systems, or deep learning training.
Compiler design and implementation.
Distributed system design, development, and deployment.
Experience with deep learning frameworks (JAX, PyTorch).
GPU training and inference of deep learning models.
C/C++ and Python programming, with excellent software design skills, including debugging, performance analysis, and test design.
Experience with parallel computing libraries (MPI).
Be able to work independently, define project goals and scope, and lead your own development efforts.
Strong communication skills are required, including working directly with engineers at partner companies to debug and tune large training runs, along with the ability to work in a dynamic product-oriented team.

Ways to stand out from the crowd:

Experience with JAX internals (jaxpr, tracing, sharding, shard_map), or with XLA, StableHLO, or MLIR.
Experience building or tuning large training jobs with pipeline, tensor, expert, and data parallelism, for example in MaxText or Megatron-LM, and debugging their performance on large GPU clusters.
Knowledge of GPU communication libraries such as NCCL, NVSHMEM, or RDMA, and of CPU and/or GPU architecture. CUDA programming experience is a plus.
Contributions to open-source compiler or machine learning projects.
Experience in mentoring early career engineers and interns is a bonus.

With competitive salaries and a generous benefits package, we are widely considered to be one of the technology world's most desirable employers. We have some of the most forward-thinking and hardworking people in the world working for us and, due to unprecedented growth, our exclusive engineering teams are rapidly growing. If you're a creative and autonomous engineer with a real passion for technology, we want to hear from you.

Your base salary will be determined based on your location, experience, and other objective job-related factors. The expected base salary range for this position is 184,000 USD – 287,500 USD. You will also be eligible for equity and benefits.

Applications for this job will be accepted at least until November 6, 2026. This posting is for an existing vacancy. NVIDIA uses AI tools in its recruiting processes.

NVIDIA is committed to fostering an inclusive work environment and proud to be an equal opportunity employer. As we highly value diversity in our current and future employees, we do not discriminate (including in our hiring and promotion practices) on the basis of race, religion, color, national origin, gender, gender expression, sexual orientation, age, marital status, veteran status, disability status or any other characteristic protected by law.

#deeplearning

Apply About employer

Report this job