Performance Engineer - Deep Learning - CA, Santa Clara, United States
$124,000 - $241,500 USD yearly
Originally posted 5 October 2026 by the employer.
About the role
This role focuses on building and optimizing libraries and tools for Deep Learning models performance engineering, particularly for PyTorch and JAX frameworks.
What you'll do
- Build and support Transformer Engine for accelerating Large Language Models.
- Collaborate on systems research to improve Deep Learning model performance, including low precision and parallelism methods.
- Implement, benchmark, and optimize new Deep Learning models such as LLMs on NVIDIA GPUs and systems.
- Build and contribute to NVIDIA submissions on community benchmarks such as MLPerf.
- Engage with the open-source community and support partners with NVIDIA's hardware and software innovations.
- Influence the design of new hardware generations and core platform software components.
What you'll need
- BS or equivalent experience in Computer Science, Electrical Engineering, or a related field.
- 2+ years of experience in C++ and Python programming.
- Strong background, experience, or coursework in parallel systems programming, preferably on GPUs.
- Knowledge of Computer Architecture, Code Optimization, and/or Operating Systems.
- Proven experience in developing large software projects.
Skills: Deep Learning, PyTorch, JAX, GPU, C++, Python
This role has been open 0 days — well below the 56-day median for Infrastructure/Platform Software roles.
Infrastructure/Platform Software · GPU Software Stack
|
Open roles in category
225
|
Median days open
56 d
|
Median salary
$236k
|
See the full market breakdown ▾Category comparison, skills in demand, and who else is hiring
| Metric | NVIDIA | All employers we track in this specialty (225 roles · 19 employers) |
|---|---|---|
| Open roles in this specialty | 127 | 225 |
| Open roles in the wider Software, Firmware & Systems family | 971 | 5996 · 146 employers |
| Median days open | 49 d | 56 d (−7 d vs this employer) |
| Median salary (USD postings) | — | $236k |
Skills observed across this category: Deep Learning, PyTorch, JAX, GPU, C++, Python
Who's hiring in this category
- NVIDIA (this employer) · 122 open roles · median 52 d
- AMD · 57 open roles · median 55 d
- Qualcomm · 17 open roles · median 165 d
- Arm Holdings · 5 open roles · median 103 d
- Intel Corporation · 5 open roles · median 53 d
- d-Matrix · 2 open roles · median 111 d
How we counted: 225 open Infrastructure/Platform Software (GPU Software Stack) roles from 19 employers tracked in the SemiconductorJobs index, counted 6 Oct 2026. Specialty figures count only roles carrying this exact specialty label, so an employer's related work in neighbouring specialties is not included there — it is counted in the wider Software, Firmware & Systems family row. Figures refresh nightly.