Senior Machine Learning Compiler Engineer

Build and maintain machine learning compiler technologies that optimize AI models from PyTorch or ONNX for CPU, GPU and NPU devices. Develop graph optimization, partitioning, debugging and validation tools, and collaborate with hardware engineers and developers to ensure models run efficiently and accurately across backends. Build and maintain compilers that translate AI models into efficient code for CPU, GPU and NPU processors. Optimize and partition model graphs for AI Hub Compiler and ONNX …