Deep Learning Computer Architect - College Grad

We are looking for a Deep Learning Computer Architect to design hardware accelerator and processor architectures that enable state‑of‑the‑art machine learning and data analytics algorithms on NVIDIA’s next‑generation mobile, embedded, and datacenter platforms.

• Contribute to features that help next‑generation GPUs advance the state of AI as a member of the deep learning architecture team.
• Stay current with the latest DL research and collaborate with DL researchers, hardware architects, and software engineers.
• Analyze the behavior of various deep learning methods, propose new features to accelerate or enable these methods, and study 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 one of the following areas:
• Computer architecture, including GPU and system level architecture
• Performance analysis and optimization
• Experience with LLM workloads, including performance tuning such as parallelization and fusion strategies
• Experience with core deep learning kernels such as matrix multiply, attention, communication, and convolution
• Programming fluency with C++ and ideally Python.
• Experience with GPU computing (CUDA).
• Experience with deep learning frameworks such as 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 – $195,500 for Level 2 and $152,000 – $241,500 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. We do not discriminate 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.