Senior ASIC Verification Engineer, GPU
NVIDIA
PerlPython
Senior Deep Learning Compiler Engineer, CUDA
NVIDIA
full-time
Posted on:
Location Type: Hybrid
Location: Redmond • California, Texas, Washington • 🇺🇸 United States
Visit company websiteSalary
💰 $184,000 - $356,500 per year
Job Level
Senior
Tech Stack
Open SourcePythonPyTorch
About the role
- Develop new ML/DL compiler abstractions, build runtimes, and ML/DL-compiler driven system solutions to accelerate large language models and other ML workloads
- Design and implement Pythonic language interface for tile-aware GPU programming
- Optimize compiler pipelines for efficient execution
- Integrate compilers with AI/ML frameworks
- Develop performance-critical primitives for tensor operations and memory operations
- Collaborate with hardware teams to co-design compiler optimizations for emerging GPU architectures, including Tensor Core utilization and distributed execution
- Build close technical relationships with internal NVIDIA software and hardware teams to push developments to NVIDIA products
Requirements
- Bachelor's degree in Computer Science, Electrical Engineering, or related field (or equivalent experience); MS or PhD are preferred
- 6+ years (academic/ industry) experience with ML/DL systems development preferable for compilers
- Strong Python and C/C++ programming skills
- Expert experience in developing or using deep learning frameworks (e.g. PyTorch, JAX, Triton, etc.)
- Strong sense of ownership, fast learner, passion for quality and user experience
- Proficiency in Python and C++ for DSL development and low-level compiler internals
- Experience with compiler frameworks (Triton, LLVM, MLIR, TVM) and IR design for GPUs
- Deep understanding of GPU architectures (CUDA cores, Tensor Cores, memory hierarchies) and tile-based execution models
- Strong experience machine learning systems research and productionization
- Open source project ownership or contributions
Benefits
- Highly competitive salaries
- Comprehensive benefits package
- Eligible for equity
Applicant Tracking System Keywords
Tip: use these terms in your resume and cover letter to boost ATS matches.
Hard skills
PythonC++ML systems developmentDL systems developmentcompiler frameworksIR designGPU architecturestile-based execution modelsdeep learning frameworksperformance optimization
Soft skills
strong sense of ownershipfast learnerpassion for qualityuser experience focus
Certifications
Bachelor's degree in Computer ScienceBachelor's degree in Electrical EngineeringMaster's degreePhD
