Nvidia Research Project Intern 

China, Beijing, Beijing 

428120634

What you’ll be doing:

• Apply your experience and knowledge in areas of AI, mathematics, accelerated computing, and relatable areas such as computer vision, machine learning.

• Conduct HER and Modulus engineering via playing with software and hardware and collaborating with external researchers. Partner with internal groups to transfer technologies and innovate products.

• Assist in building neural operators in modulus and FP8 optimization on GPU. With the state-of-the-art AI models, you can even create an new solver for some scientific function or optimize the AI models with FP8 on GPUs. It can also help you use multiple SDKs or frameworks to realize your idea!

• Mainly works in Beijing office, 3~4 days/week with at least 6 months available.

What we need to see:

• You are pursuing a Masters in Computer Science, Mathematics, Electrical Engineering, or a related discipline.

• Programming experience and proficiency in Python, C/C++, and familiar with Linux developing environment.

• Ability to communicate effectively in a collaborative environment.

Ways to stand out from the crowd:

• Have knowledge on developing on PINN operators or doing FP8 optimizations.

• Experience and/or theoretical knowledge in computer vision, PINN and Artificial Intelligence.

• Experience in CUDA and deep learning frameworks (i.e TensorFlow or PyTorch).

• Knowledge of using DNN to solve PDE or brain-inspired computing background.