Skip to content
ITA Jobs
All vacancies
NVIDIA logo

Senior High-Performance LLM Training Engineer

NVIDIA

US, CA, Santa Clara · On-site · relocation$15,333–29,708 gross/mofull-timeNVIDIA Careers (offices abroad)

NVIDIA is seeking experienced engineers specializing in performance analysis and optimization to improve the efficiency of LLM training workloads, which are shaping the world's most advanced computing systems. This position focuses on optimizing NVIDIA’s high-performance LLM software stack in frameworks like PyTorch and JAX for high-performance training on thousands of GPUs, while also helping shape hardware roadmaps for the next generation of GPUs powering the AI revolution.

Responsibilities

  • Implement production-quality software in multiple layers of NVIDIA's deep learning platform stack, from drivers to DL frameworks.
  • Build and support NVIDIA submissions to the MLPerf Training benchmark suite.
  • Implement key DL training workloads in NVIDIA's proprietary processor and system simulators to enable future architecture studies.
  • Build tools to automate workload analysis, workload optimization, and other critical workflows.

Languages

    Work format
    On-site
    Seniority
    Senior
    Posted
    8 Apr 2026
    Last verified
    In the last 5 days

    Similar roles

    Keep looking