About the job
We are now looking for a Senior 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.
Responsibilities
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.
Qualifications
Minimum
MS or PhD degree in computer science, computer architecture, electrical engineering or related field or equivalent experience.
5+ years of relevant experience in at least a few of the following relevant areas is required in your work history:
- 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
Preferred
Experience with GPU computing (CUDA)
Experience with deep learning frameworks like PyTorch