Senior Machine Learning Engineer, AI, SIML

Apple's ISE team is seeking a senior machine learning engineer to design scalable training and inference systems for Generative AI. This role involves developing core components for Apple's ML platform, pushing limits of training technologies, and deploying work across Apple Intelligence products. Opportunities to open-source contributions are available.

• Develop core components for Apple's scalable ML platform
• Push limits of existing training technologies
• Create new techniques to overcome system constraints
• Deploy work on high-impact tasks across teams building Apple Intelligence products
• Open-source contributions

• PhD or Master's in CS or equivalent industry experience with 3+ years in AI/ML
• Strong Python programming skills
• Solid understanding of software-hardware co-design principles and algorithms
• Solid understanding of PyTorch software stack and experience maintaining ML frameworks
• Strong understanding of LLM architectures and core building blocks

• Experience on AI/ML-optimized runtime stacks
• Familiarity with parallelization algorithms for large model training
• Familiarity with recent developments in foundation model architectures