AIML Researcher: Foundation Models & Private AI

Apple Inc.; Cupertino (CA); On-site; USD 150,000 - 225,000; Full time

Benefits offered by this job

Medical and dental coverage

Retirement benefits

Apple stock programs

Tuition reimbursement

Job summary

Apple's Machine Learning and AI team in Cupertino, California is building frontier foundation models optimized for Apple silicon, privacy, and deep OS integration.

You will work across the full training lifecycle—including pre-training foundation models and developing mid-training approaches that bridge general capability and task-specific performance—addressing reward modeling, sparse and delayed rewards, and alignment challenges for real-world experiences.

Qualifications

• Deep learning expertise with publications or strong product applications.
• Python programming proficiency with JAX, PyTorch, or TensorFlow.
• Collaborative, team-oriented work style.
• PhD or equivalent practical experience in Computer Science or related field.

Skills

Deep learning expertise

Python programming

Collaborative work

Education

PhD or equivalent

Tools

JAX

PyTorch

TensorFlow

Job description

Apple's Machine Learning and AI team in Cupertino, California is building frontier foundation models optimized for Apple silicon, privacy, and deep OS integration.

You will work across the full training lifecycle—including pre-training foundation models and developing mid-training approaches that bridge general capability and task-specific performance—addressing reward modeling, sparse and delayed rewards, and alignment challenges for real-world experiences.