AIML - ML Engineer, Responsible AI

Apple is seeking a Machine Learning Engineer to work on responsible AI, focusing on safety, robustness, and uncertainty in generative models. You will develop models, tools, and metrics for evaluating safety, and build human annotation and red teaming pipelines for various Apple products. This role involves working with large generative models for vision and language, and interpreting failures in language and diffusion models.

• Develop models, tools, metrics, and datasets for assessing and evaluating the safety of generative models over the model deployment lifecycle
• Develop methods, models, and tools to interpret and explain failures in language and diffusion models
• Build and maintain human annotation and red teaming pipelines to assess quality and risk of various Apple products
• Prototype, implement, and evaluate new ML models and algorithms for red teaming LLMs

• Strong engineering skills and experience in writing production-quality code in Python, Swift or other programming languages
• Background in generative models, natural language processing, LLMs, or diffusion models
• Experience with failure analysis, quality engineering, or robustness analysis for AI/ML based features
• Experience working with crowd-based annotations and human evaluations
• Experience working on explainability and interpretation of AI/ML models