Back to jobs

Deep Learning Performance Architect

NVIDIA
China, Shanghai
Full-time
You apply on NVIDIA's own careers site

This architect analyzes deep-learning workloads across hardware and software architectures to identify performance bottlenecks. The work supports decisions about architecture and system configurations that improve performance and energy efficiency.

Permanent
3+ years preferred
Bachelor's, Master's, or PhD

Skills & Expertise

Deep learning performance analysis
GPU architecture
Computer architecture
System architecture design
Performance optimization
Machine learning frameworks

Key Responsibilities

Analyze deep-learning workload performance across different architectures.

Identify software and architecture performance bottlenecks.

Evaluate features and system configurations for performance and energy efficiency.

Full Description

NVIDIA is developing GPU and system architectures that accelerate deep learning and high-performance computing applications. We are looking for an expert deep learning performance architect to join our deep learning modelling, performance projections, analysis and optimization effort. In this position, you will have the chance to analyze deep learning performance on different hardware and software architecture and make a significant impact in a dynamic technology focused company

What you’ll be doing:

• Analyze performance of various deep learning workloads on different architectures

• Identify architecture and software performance bottlenecks

• Explore new features and system configurations to achieve better performance and energy efficiency

What we need to see:

• BSc. MS or PhD in relevant discipline (CS, EE, Math, etc.,)

• 3+ years of working experience in relevant directions (e.g., hardware system, datacenter hardware, LLM workloads on data center) will be a plus

• Be familiar with GPU or accelerator-based deep learning platform and software stack

• A strong background in computer architecture

• Experience on system architecture design and performance optimization

• Familiar with machine learning and deep learning frameworks

Applications are handled on NVIDIA's site