Senior Machine Learning Compiler Engineer
NVIDIA
Python
GPU Architecture Engineer
NVIDIA
full-time
Posted on:
Origin: • 🇺🇸 United States • California, North Carolina
Visit company websiteSalary
💰 $120,000 - $235,750 per year
Job Level
Mid-LevelSenior
Tech Stack
Python
About the role
- Develop and enhance various features in the GPU architecture.
- Contribute to advancing GPU Architecture Simulators, GPU testing infrastructure and metrics.
- Understand and model in C++ and python features for future GPU graphics and parallel processing architectures.
- Develop test plans and testing infrastructure.
- Debug tests on architecture simulators.
- Develop tools to validate performance models and verify functional models.
Requirements
- Pursuing or recently completed an MS or PhD (or equivalent experience) in CS, EE, math or related discipline.
- Strong programming ability in C, C++, and Python.
- Deep knowledge of computer architecture and parallel programming.
- Experience in machine learning/deep learning and/or graphics algorithms, and/or signal processing.
- Prior architecture experience.
- Programming in CUDA and/or DX and Vulkan.
