Description

Join NVIDIA's Developer Technology team to work on AI and deep learning optimization for GPU architectures. Collaborate with various teams to influence next-gen software platforms.

Duration

3 months

Academic Requirement

Pursuing an MS or PhD in an engineering or Computer Science related discipline or equivalent program area.

Responsibilities

• Study and develop state of the art techniques in deep learning and machine learning.
• Perform in-depth analysis and optimization for GPU architectures.
• Collaborate with architecture, research, libraries, tools, and system software teams.

Requirements

• Strong knowledge of C/C++, software development, and AI algorithms.
• Experience with parallel programming and performance optimizations, ideally CUDA.
• Mathematical fundamentals, including linear algebra.
• Experience with Deep Learning models through frameworks like PyTorch, JAX, DGL, PyG.
• Excellent communication and organization skills.
• Experience with low-precision arithmetic.