Developer Technology Engineer, AI

at NVIDIA · AI Frontier · Shanghai, China +1

NVIDIA Developer Technology Engineer focused on optimizing core parallel algorithms and data structures for GPUs, specifically working with LLM training frameworks and performance optimization. Collaborates with application developers and internal NVIDIA teams to improve performance and developer efficiency.

What you'd actually do

  1. Working directly with key application developers to understand the current and future problems they are solving, crafting and optimizing core parallel algorithms and data structures to provide the best solutions using GPUs, through both reference code development and direct contribution to the applications.
  2. Collaborating closely with diverse groups at_ _NVIDIA_ _such as the architecture, research, libraries, tools, and system software teams to influence the design of next-generation architectures, software platforms, and programming models, by investigating the impact on application performance and developer efficiency.
  3. Need to travel from time to time for conferences and for on-site visits with developers.

Skills

Required

  • BS, MS, or PhD degree from a leading university in an engineering or computer science related field (or equivalent experience)
  • 3+ years experience, programming proficiency in C/C++ and/or Python with a deep understanding of software design, programming techniques, and algorithms.
  • 2+ years experience with LLM training framework development and performance optimization.
  • Strong mathematical fundamentals, including linear algebra and numerical methods.
  • Experience with parallel programming, ideally CUDA C/C++.
  • Strong communication and organization skills, with a logical approach to problem solving, good time management, and task prioritization skills.

Nice to have

  • domain expertise in LLM, CTR, CV, or HPC

What the JD emphasized

  • LLM training framework development and performance optimization

Other signals

  • GPU optimization
  • parallel algorithms
  • LLM training framework development
  • performance optimization
Read full job description

NVIDIA has been redefining computer graphics, PC gaming, and accelerated computing for more than 30 years. It’s a unique legacy of innovation that’s fueled by great technology—and amazing people. Today, we’re tapping into the unlimited potential of AI to define the next era of computing. An era in which our GPU acts as the brains of computers, robots, and self-driving cars that can understand the world. Doing what’s never been done before takes vision, innovation, and the world’s best talent. As an NVIDIAN, you’ll be immersed in a diverse, supportive environment where everyone is inspired to do their best work. Come join the team and see how you can make a lasting impact on the world.

What You'll Be Doing:

  • Working directly with key application developers to understand the current and future problems they are solving, crafting and optimizing core parallel algorithms and data structures to provide the best solutions using GPUs, through both reference code development and direct contribution to the applications.
  • Collaborating closely with diverse groups at_ NVIDIA _such as the architecture, research, libraries, tools, and system software teams to influence the design of next-generation architectures, software platforms, and programming models, by investigating the impact on application performance and developer efficiency.
  • Need to travel from time to time for conferences and for on-site visits with developers.

What We Need To See:

  • A BS, MS, or PhD degree from a leading university in an engineering or computer science related field (or equivalent experience). While not a requirement, domain expertise in LLM, CTR, CV, or HPC, is helpful.
  • 3+ years experience, programming proficiency in C/C++ and/or Python with a deep understanding of software design, programming techniques, and algorithms.
  • 2+ years experience with LLM training framework development and performance optimization.
  • Strong mathematical fundamentals, including linear algebra and numerical methods.
  • Experience with parallel programming, ideally CUDA C/C++.
  • Strong communication and organization skills, with a logical approach to problem solving, good time management, and task prioritization skills.

Widely considered to be one of the technology world’s most desirable employers, NVIDIA offers highly competitive salaries and a comprehensive benefits package. As you plan your future, see what we can offer to you and your family www.nvidiabenefits.com/

NVIDIA is committed to fostering a diverse 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