Expoint - all jobs in one place

Finding the best job has never been easier

Limitless High-tech career opportunities - Expoint

Nvidia Software Engineer cuDNN - Deep Learning 
China, Shanghai 
381775305

24.06.2024

What you’ll be doing:

  • Develop production-quality software that ships as part of NVIDIA's AI software stack, including optimized large language model (LLM) support.

  • Analyze the performance of important workloads, tuning our current software, and proposing improvements for future software.

  • Work with cross-collaborative teams of deep learning software engineers and GPU architects to innovate across applications like generative AI, autonomous driving, computer vision, and recommender systems.

  • Adapt to the constantly evolving AI industry by being agile and excited to contribute across the codebase, including API design, software architecture, performance modeling, testing, and GPU kernel development.

  • Mentoring junior engineers on the team.

What we need to see:

  • M.S. degree in computer science (or similar) or equivalent experience.

  • Strong programming skills in C/C++ development, work experience with CUDA development, and familiarity with Python.

  • Good understanding of linear algebra.

  • Familiarity with the latest trends in machine learning.

  • Experience designing high level software architecture.

  • Experience with performance analysis, profiling, and code optimization.

  • Ability to work independently, define project goals and scope, and lead your own development effort.

Ways to stand out from the crowd:

  • GPU programming and optimization expertise (e.g. CUDA or OpenCL).

  • Practical experience with machine learning, especially deep learning.

  • Experience with computer architecture and building performance models for CPUs, GPUs, or other accelerators.

  • Excellent problem solving skills, including applications of algorithms and data structures.

  • Strong experience with data science, statistical analysis, and visualization.