Used Tools & Technologies
Not specified
Required Skills & Competences ?
Python @ 7 Algorithms @ 4 Machine Learning @ 4 PyTorch @ 4 CUDA @ 3 GPU @ 7Details
NVIDIA is seeking a Senior Deep Learning Performance Architect with expertise in performance analysis, performance modeling, and AI/deep learning. This role focuses on developing next-generation architectures to accelerate AI and high-performance computing applications.
Responsibilities
- Develop innovative architectures to advance deep learning performance and efficiency.
- Analyze performance, cost, and power trade-offs by developing analytical models, simulators, and test suites.
- Understand and analyze hardware and software architecture interplay on future algorithms, programming models, and applications.
- Develop, analyze, and utilize deep learning frameworks, libraries, and compilers.
- Collaborate with software, product, and research teams to guide deep learning hardware and software direction.
Requirements
- MS or PhD in Computer Science, Computer Engineering, Electrical Engineering, or equivalent experience.
- 6+ years of relevant work experience.
- Strong background in GPU or Deep Learning ASIC architecture for training and/or inference.
- Experience with performance modeling, architecture simulation, profiling, and analysis.
- Solid foundation in machine learning and deep learning.
- Strong programming skills in Python, C, and C++.
Ways to Stand Out
- Experience with deep neural network training, inference, and optimization in frameworks such as PyTorch, JAX, TensorRT.
- Familiarity with libraries, compilers, and languages: cuDNN, cuBLAS, CUTLASS, MLIR, Triton, CUDA, OpenCL.
- Experience with other deep learning accelerator architectures or workload analysis.
- Demonstrated self-motivation and critical thinking skills.
Benefits
Base salary range is $184,000 - $356,500 USD, determined by location, experience, and similar employee pay. Eligible for equity and benefits. NVIDIA is an equal opportunity employer committed to diversity.