Machine Learning Compiler Developer at NVIDIA

Join NVIDIA as a Machine Learning Compiler Developer, focusing on LPX inference and compiler optimizations. This role offers the chance to shape how neural network tasks are executed efficiently on future platforms.
In this position, you will develop algorithms for high-performance applications while collaborating with esteemed teams in hardware and software. The role requires hands-on experience with compiler development and familiarity with tools such as LLVM and deep learning frameworks like TensorFlow and PyTorch.
Key Responsibilities:
• Create optimized runtime and compiler components for inference
• Map workloads on NVIDIA systems for enhanced performance
• Collaborate across teams to influence architectural design
• Monitor performance metrics to ensure efficiency
• Prototype new techniques for compiler optimizations
Requirements:
• MS or PhD in a relevant field or equivalent experience
• Experience in systems-level programming
• Knowledge of com...