Foundation Models ML Engineer - On-Device Private AI

Apple Inc.; Cupertino (CA); On-site; USD 181,100 - 318,400; Full time

Benefits offered by this job

Comprehensive medical and dental coverage

Retirement benefits

Employee stock purchase program

Tuition reimbursement for education

Job summary

Apple Inc. is looking for a Machine Learning Engineer, Foundation Models to join their Cupertino team. This role involves developing novel applications of deep learning that enhance user experiences across Apple products, working with world-class engineers and scientists on cutting-edge challenges.

Ideal candidates will hold a PhD or equivalent experience in Computer Science and be proficient in Python, with skills in deep learning toolkits like JAX or PyTorch. Apple offers comprehensive benefits including stock options and educational reimbursements.

Qualifications

• Proven track record in training or deployment of large models or building large-scale distributed systems.
• Ability to work in a collaborative environment.

Responsibilities

• Identify and develop novel applications of deep learning in Apple products.
• Work with a fast-growing team to tackle challenging problems in foundation models and deep learning.

Skills

Proficient programming skills in Python

Experience with deep learning toolkits (JAX, PyTorch, TensorFlow)

Collaborative working abilities

Education

PhD or equivalent practical experience in Computer Science

Tools

JAX

PyTorch

TensorFlow

Job description

Apple Inc. is looking for a Machine Learning Engineer, Foundation Models to join their Cupertino team. This role involves developing novel applications of deep learning that enhance user experiences across Apple products, working with world-class engineers and scientists on cutting-edge challenges.

Ideal candidates will hold a PhD or equivalent experience in Computer Science and be proficient in Python, with skills in deep learning toolkits like JAX or PyTorch. Apple offers comprehensive benefits including stock options and educational reimbursements.