Nvidia Compute Architect 

China, Shanghai 

974782045

What You'll Be Doing:

• Design, implement, and optimize MLIR-based compiler passes for deep learning and data analytics workloads.

• Analyze and improve the performance of machine learning and deep learning algorithms on current and next-generation architectures with compiler technologies.

• Identify performance bottlenecks in compiler-generated code and propose creative solutions.

• Collaborate with hardware architects and software teams to co-design features that maximize performance and efficiency.

• Contribute to the evolution of NVIDIA’s deep learning compiler stack and libraries.

What We Need to See:

• MS or PhD in Computer Science, Electrical Engineering, Mathematics, or a related field, or equivalent experience.

• 5+ years of working experience

• Proven experience developing compilers or compiler infrastructure, preferably with MLIR, LLVM, or similar frameworks.

• Strong programming skills in C++ and Python.

• Solid understanding of computer architecture, especially as it relates to performance optimization.

• Experience optimizing code for CPUs or GPUs, including low-level programming (assembly, SIMD, or vectorization).

• Experience with deep learning algorithms, especially matrix multiplication and convolution.

Ways to Stand Out from the Crowd:

• Hands-on experience with MLIR, LLVM, or other modern compiler frameworks.

• Deep understanding of parallel programming models and GPU architectures.

• Strong communication and organizational skills.

• Demonstrated ability to work collaboratively in a fast-paced, cross-functional environment.