Senior AI Compiler Engineer

Job Summary

Joining a dynamic team, the full-time Senior AI Compiler Engineer will analyze deep learning networks, develop compiler optimization algorithms, and enhance performance for NVIDIA's Deep Learning & AI Compiler, with opportunities for remote work available.

Key responsibilities

• Analyze deep learning networks and develop compiler optimization algorithms
• Implement performance optimizations and define public APIs for AI workloads
• Utilize strong programming skills in CUDA to analyze and debug performance bottlenecks on GPUs

Required qualifications

• Bachelor's, master's, or Ph.D. in Computer Science, Computer Engineering, or a related field
• 3+ years of relevant work or research experience in performance analysis and compiler optimizations
• Experience with compiler technologies such as MLIR, LLVM, XLA, or Triton
• Excellent programming skills in C/C++ and Python, including debugging and performance analysis
• Ability to work independently and lead development efforts