Deep Learning Performance Software Engineer

Reposted 8 Days Ago
Be an Early Applicant
Shanghai, Shanghai Municipality, Shanghai
In-Office
Mid level
Artificial Intelligence • Computer Vision • Hardware • Robotics • Metaverse
The Role
Develop compilers and optimized kernels for deep learning, enhance compiler architecture, and analyze performance of AI workloads.
Summary Generated by Built In

We are now looking for a Deep Learning Performance Software Engineer!

We are expanding our research and development for deep learning. We seek excellent Software Engineers and Senior Software Engineers to join our team. We specialize in developing GPU-accelerated Deep learning software. Researchers around the world are using NVIDIA GPUs to power a revolution in deep learning, enabling breakthroughs in numerous areas. Join the team that builds software to enable new solutions. Your ability to work in a fast-paced customer-oriented team is required and excellent communication skills are necessary. 

What you’ll be doing:

  • Develop compilers and DSLs for deep learning workloads

  • Design and implement highly optimized deep learning kernels

  • Continuously improve the compiler architecture for current and next generation chips

  • Perform performance analysis on emerging AI workloads and integrate with AI frameworks

What we need to see: 
  • Master's or Ph.D degree (or equivalent experience)  in relevant discipline (CE, CS&E, CS, AI)

  • Excellent C/C++ programming and software design skills

  • Experience with XLA, TVM, MLIR, LLVM, deep learning models and algorithms

  • 3+ years of relevant work experience

NVIDIA is widely considered to be one of the technology world’s most desirable employers. We have some of the most brilliant and talented people on the planet working for us. If you're creative and autonomous, we want to hear from you!

Top Skills

C/C++
Llvm
Mlir
Tvm
Xla
Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
HQ: Santa Clara, CA
21,960 Employees
Year Founded: 1993

What We Do

NVIDIA’s invention of the GPU in 1999 sparked the growth of the PC gaming market, redefined modern computer graphics, and revolutionized parallel computing. More recently, GPU deep learning ignited modern AI — the next era of computing — with the GPU acting as the brain of computers, robots, and self-driving cars that can perceive and understand the world. Today, NVIDIA is increasingly known as “the AI computing company.”

Similar Jobs

NVIDIA Logo NVIDIA

Software Engineer

Artificial Intelligence • Computer Vision • Hardware • Robotics • Metaverse
In-Office
Shanghai, Shanghai Municipality, Shanghai, CHN
21960 Employees

NVIDIA Logo NVIDIA

Software Engineer

Artificial Intelligence • Computer Vision • Hardware • Robotics • Metaverse
In-Office
2 Locations
21960 Employees

Adyen Logo Adyen

Team Lead

Fintech • Payments • Financial Services
Easy Apply
Hybrid
Shanghai, Shanghai Municipality, Shanghai, CHN
4568 Employees

BlackRock Logo BlackRock

VP, Information Security

Fintech • Information Technology • Financial Services
In-Office
Shanghai, Shanghai Municipality, Shanghai, CHN
25000 Employees

Similar Companies Hiring

Idler Thumbnail
Artificial Intelligence
San Francisco, California
6 Employees
Fairly Even Thumbnail
Software • Sales • Robotics • Other • Hospitality • Hardware
New York, NY
Bellagent Thumbnail
Artificial Intelligence • Machine Learning • Business Intelligence • Generative AI
Chicago, IL
20 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account