Senior Performance Engineer - Deep Learning

NVIDIA
Santa Clara
Workplace: OnsiteFull timeUSD 152,000 - 241,500 annuallyFunction: Data Science & Machine LearningExperience: 3+ yearsEducation: bachelorsSkills: ["Problem-solving","Communication","Collaboration"]

Senior Performance Engineer - Deep Learning at NVIDIA focused on building and optimizing libraries and tools to accelerate deep learning workloads. You will contribute to open-source DL frameworks, optimize transformer workflows, and scale LLMS on NVIDIA GPUs, collaborating across teams and with the open-source community to push performance, efficiency, and hardware-software integration.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
NVIDIA
NVIDIA
5 months ago

Senior Performance Engineer - Deep Learning

âś“ Verified Job

Canonical indexed version, validated from employer's careers page.

Source: Company careers pageValidated by: Fursa AI
Last checked: 9 minutes agoStatus: Live

Job Summary

Senior Performance Engineer - Deep Learning at NVIDIA focused on building and optimizing libraries and tools to accelerate deep learning workloads. You will contribute to open-source DL frameworks, optimize transformer workflows, and scale LLMS on NVIDIA GPUs, collaborating across teams and with the open-source community to push performance, efficiency, and hardware-software integration.
Location: Santa Clara
Workplace: Onsite
Employment Type: Full time
Job Function: Data Science & Machine Learning

Key Responsibilities

  • •Build and support Transformer Engine, the open-source library for accelerating the training of Large Language Models.
  • •Collaborate on systems research that improves Deep Learning model performance, such as training using extremely low precision, parallelism methods, etc.
  • •Implement, benchmark, and optimize new Deep Learning models such as LLMs straight out of groundbreaking research to scale efficiently on NVIDIA GPUs and systems.
  • •Build and contribute to NVIDIA submissions on community benchmarks such as MLPerf.
  • •Engage with the open-source community as well as support enterprise customers and partners by delivering the benefits of NVIDIA’s latest hardware and software innovations.

Pay and Benefits

Salary: USD 152,000 - 241,500 annually
Equity and Bonus:Equity
Perks:Equity

Key Requirements

  • •BS or equivalent experience in Computer Science, Electrical Engineering, or a related field.
  • •3+ 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.
Experience:3+ yearsSoftwareDeep learningOpen source
Education:Bachelor's
Skills:Problem-solvingCommunicationCollaboration
Languages:English
Tech Stack:C++PythonPyTorchJAXCUDACuDNNCuBLASCuSOLVEROpenAI TritonCuTeDSLPallasGPUMLPerf

Company Brief

NVIDIA
Designs and manufactures GPUs, AI accelerators, and system-on-chip products for gaming, data centers, professional visualization, and automotive markets, enabling advanced graphics, AI, and high-performance computing solutions worldwide.
Industry: Electronics Manufacturing
Company Size: Enterprise (1,001+ employees)
Revenue: USD 1B+
Growth: Public Company
Valuation: Public Company (Market Cap in USD)
Funding: IPO / Publicly Listed
Headquarters: Santa Clara, United States
Founded: 1993
Glassdoor
Glassdoor: 4.3
WebsiteLinkedInGlassdoor