NVIDIA

Software Engineer, cuDNN - Deep Learning

Beijing, China Shanghai, China
Python Machine Learning Deep Learning API
Description

Do you love writing fast code and crafting software systems to solve complex problems? We are looking for hardworking software engineers to help design, build, and ship cuDNN: our GPU-accelerated library of primitives for deep neural networks. Intelligent machines powered by AI computers that can learn, reason, and interact with people are no longer science fiction. This is truly an extraordinary time. The era of AI has begun, and we are powering it. If this role seems like a good match for your skills and interests, tell us why you think you might be a great fit for our team, and we'd love to tell you more about what we do!

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.

There are more than 50,000 engineering jobs:

Subscribe to membership and unlock all jobs

Engineering Jobs

60,000+ jobs from 4,500+ well-funded companies

Updated Daily

New jobs are added every day as companies post them

Refined Search

Use filters like skill, location, etc to narrow results

Become a member

🥳🥳🥳 452 happy customers and counting...

Overall, over 80% of customers chose to renew their subscriptions after the initial sign-up.

To try it out

For active job seekers

For those who are passive looking

Cancel anytime

Frequently Asked Questions

  • We prioritize job seekers as our customers, unlike bigger job sites, by charging a small fee to provide them with curated access to the best companies and up-to-date jobs. This focus allows us to deliver a more personalized and effective job search experience.
  • We've got about 70,000 jobs from 5,000 vetted companies. No fake or sleazy jobs here!
  • We aggregate jobs from 5,000+ companies' career pages, so you can be sure that you're getting the most up-to-date and relevant jobs.
  • We're the only job board *for* software engineers, *by* software engineers… in case you needed a reminder! We add thousands of new jobs daily and offer powerful search filters just for you. 🛠️
  • Every single hour! We add 2,000-3,000 new jobs daily, so you'll always have fresh opportunities. 🚀
  • Typically, job searches take 3-6 months. EchoJobs helps you spend more time applying and less time hunting. 🎯
  • Check daily! We're always updating with new jobs. Set up job alerts for even quicker access. 📅

What Fellow Engineers Say