AI Compiler Engineer

Darwin Recruitment is hiring an AI Compiler Engineer for an international semiconductor company developing high-performance AI processors for intelligent edge systems. The role involves building compiler technology for custom ML accelerators, translating neural-network models into efficient hardware-executable workloads, and optimizing computational graphs. Candidates should have strong C++ and systems fundamentals, with a Master's degree in a related field and knowledge of compiler design and computer architecture. The position is on-site in San Jose, USA.

• Build and improve compiler technology for custom machine-learning accelerators
• Translate neural-network models into efficient hardware-executable workloads
• Develop compiler passes covering graph transformation, operator lowering, scheduling and code generation
• Optimise computational graphs through techniques such as operator fusion, constant folding and layout transformation
• Support model ingestion from frameworks and formats such as PyTorch, TensorFlow, ONNX and related ecosystems
• Analyse workloads to identify compiler, runtime and hardware bottlenecks
• Work closely with processor architects to enable new instructions and hardware capabilities
• Collaborate with kernel, runtime, driver and AI framework engineers

• Master’s degree in Computer Science, Computer Engineering, Electrical Engineering or a related technical discipline