Deep Learning Computer Architect - College Grad

We are now looking for a Deep Learning Computer Architect! NVIDIA is seeking architects like you to help design hardware accelerator and processor architectures that enable state of the art machine learning and data analytics algorithms and applications on our next‑generation mobile, embedded and datacenter platforms. This position offers you the opportunity to have a real impact in a dynamic, technology‑focused company.

• As a member of our deep learning architecture team, you will contribute to features that help next‑generation GPUs advance the state of AI.
• This position requires you to keep up with the latest DL research and collaborate with diverse teams (internal and external to NVIDIA), including DL researchers, hardware architects, and software engineers.
• Your day‑to‑day work will include analyzing the behavior of various deep learning methods, proposing new features to accelerate or enable various methods, and studying the benefits of the proposed features.

• MS or PhD degree in computer science, computer architecture, electrical engineering, or related field, or equivalent experience.
• 2+ years of relevant experience in at least a few of the following areas:
• Computer architecture, including GPU and system level architecture;
• Performance analysis and optimization;
• Experience with LLM workloads, including performance tuning considerations such as parallelization and fusion strategies;
• Experience with core deep learning kernels such as matrix multiply, attention, and communication convolution
• Programming fluency with C++ and ideally Python.
• Experience with GPU computing (CUDA).
• Experience with deep learning frameworks like PyTorch.

Your base salary will be determined based on your location, experience, and the pay of employees in similar positions. The base salary range is 124,000 USD – 195,500 USD for Level 2, and 152,000 USD – 241,500 USD for Level 3. You will also be eligible for equity and benefits.

NVIDIA is committed to fostering an inclusive work environment and proud to be an equal opportunity employer. As we highly value diversity in our current and future employees, we do not discriminate (including in our hiring and promotion practices) on the basis of race, religion, color, national origin, gender, gender expression, sexual orientation, age, marital status, veteran status, disability status or any other characteristic protected by law.