NVIDIA

Devtech Compute Engineer

NVIDIA  •  Shanghai, CN (Onsite)  •  3 hours ago
Apply
AI can make mistakes so check important info. Chat history is never stored.
54
AI Success™

Job Description

We’re working on the next generation of recommendation tools and pushing the boundaries of accelerating model training and inference on GPU. You’ll join a team of ML, HPC and Software Engineers and Applied Researcher developing a framework designed to make the productization of GPU-based recommender systems as simple and fast as possible.

What you'll be doing:

  • In your role as Devtech Compute Engineer or CUDA Performance Engineer you will be primarily for the development of performance critical code of our deep learning applications with the goal of establishing world class performance for our customers.

  • This includes investigating the current performance and exploring optimization opportunities together with the global developers.

  • Important part of the work is that once optimal performance has been demonstrated that these solutions are integrated into our open source software libraries like ACCV-Lab, Recsys-Example .

  • With the knowledge to the requirements from customers and performance bottleneck, you will also work with our GPU, CPU, Network team to define the next generation hardware and software solutions.

  • Our coverage is wide including: LLM, Recsys, Robotic, Assisted Driving.

What we need to see:

  • 2+ years of experience of c++ code development in collaborative software development projects

  • Skilled at writing CUDA kernels and optimizing code

  • Basic knowledge of ML algorithms and deep learning

  • Basic knowledge and understanding of mathematical topics including linear algebra, calculus and statistics

  • Experience with algorithms and optimization

  • Python and jupyter notebook for analysis, algorithm exploration and processing

  • High standard for code quality and rigorous testing practices

  • Conversational level English proficiency

  • Some experience with Linux, openMP and MPI

Way to stand out from the crowd:

  • Experience in c++ HPC code development / PhD in related fields

  • Able to perform in-depth performance analysis, can demonstrate to model the performance with mathematical and statistical considerations

  • Linear algebra, calculus and statistics as second nature and this is reflected in your background of mathematics, physics, applied science or HPC related field

  • Demonstrate the ability to write CUDA kernels with the purpose of utilizing the hardware to its full potential.

  • Write unit tests and validate the correctness of the optimizations as well as strive for and propose optimal solutions and ambitious goals, convince and help others to do the same

#deeplearning

NVIDIA

About NVIDIA

Since its founding in 1993, NVIDIA (NASDAQ: NVDA) has been a pioneer in accelerated computing. The company’s invention of the GPU in 1999 sparked the growth of the PC gaming market, redefined computer graphics, ignited the era of modern AI and is fueling the creation of the metaverse. NVIDIA is now a full-stack computing company with data-center-scale offerings that are reshaping industry.

Industry
Hardware & Semiconductors
Company Size
10,000+ employees
Headquarters
Santa Clara, CA
Year Founded
1993
Social Media