AIML - ML Engineer, Safety & Red Teaming

$175,800 - $264,200/Yr

Apple - San Francisco, CA

Full-time - Mid Level

San Francisco, CA

Computer and Electronic Product Manufacturing

About the position

The role involves working on innovative projects in generative AI at Apple, focusing on developing models, tools, and metrics to assess the safety and robustness of generative models. As part of the Apple HCMI group, the position emphasizes tackling challenges in machine learning, particularly in the areas of vision and language, while ensuring quality and risk assessment of various Apple products.

Responsibilities

• Develop models, tools, metrics, and datasets for assessing and evaluating the safety of generative models over the model deployment lifecycle.
• Develop 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.

Requirements

• 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.
• Ability to work with highly-sensitive content with exposure to offensive and controversial content.

Nice-to-haves

• BS, MS or PhD in Computer Science, Machine Learning, or related fields or an equivalent qualification acquired through other avenues.
• Proven track record of contributing to diverse teams in a collaborative environment.

Benefits

• Comprehensive medical and dental coverage
• Retirement benefits
• Discounted products and free services
• Reimbursement for certain educational expenses, including tuition
• Discretionary bonuses or commission payments
• Relocation assistance