Requirements

• Bachelors degree in an aligned field and some industry experience
• Experience with compiler development
• Excellent programming skills in C++, C or Swift
• Proficiency in English
• Strong written and verbal communication skills

Nice to Haves

• MS or PhD in computer science, machine learning, or related field
• Experience in low-level software architecture, modeling, or validation
• Familiarity with AI networks such as CNN, transformer, and diffusion model architectures and their performance characteristics
• Proficiency with profiling and optimizing complex software
• Experience with SoC low-level software development, distributed computing, or hardware acceleration
• Strong background in data science and statistical methods, with demonstrated ability to analyze large datasets and present complex insights clearly
• Experience with MLIR
• Experience in building debugging and performance analysis tooling

What You'll Be Doing

• Analyze, design, and implement software interfacing with the hardware for AI applications across Apple product platforms
• Assume technical leadership, mentor, and guide other developers
• Ensure technical excellence and robust, maintainable software
• Design and develop compiler optimizations to improve latency/bandwidth of ML models
• Develop software across compiler, framework, driver, and firmware
• Collaborate closely with model teams to support new ML models on hardware
• Conduct performance profiling and diagnostics using and developing specialized tools
• Engage with QA teams