- Shanghai, Shanghai, China
- Full-time · Senior
- 5+ years
Research, design, and implement performance optimization strategies for AI, scientific, and HPC workloads on NVIDIA Grace/Vera CPUs.
Compensation
With highly competitive salaries, a comprehensive benefits package, and a great company culture
Responsibilities
- Collaborate with developers, researchers, and framework maintainers across industries to identify and resolve performance challenges in diverse workloads such as AI, data analytics, simulation, and numerical computing.
- Profile, analyze, and optimize CPU performance from application-level algorithms down to low-level microarchitecture.
- Contribute to open-source frameworks, key software stacks, reference implementations, and performance libraries to unlock full CPU potential.
- Work closely with NVIDIA’s architecture, research, libraries, tools, and system software teams to improve our overall platform performance.
- Provide insights that shape next-generation CPU designs, compiler toolchains, and development workflows for better developer productivity and throughput.
Requirements
- BS, MS, or PhD in Computer Science, Computer Engineering, or a related field
- Nice to have: AI or data preprocessing pipelines
- Nice to have: HPC applications
- Nice to have: parallel computing
- Nice to have: distributed runtime environments