ML Engineer - Large-Scale LLMs & Multimodal AI

Apple, Inc.; Sunnyvale (CA); On-site; USD 216,000 - 325,000; Full time

Benefits offered by this job

Medical and dental coverage

Retirement benefits

Employee stock programs

Relocation assistance

Job summary

Apple in Sunnyvale, CA is seeking a senior ML engineer to help design and deploy large-scale ML systems powering Foundation Models. You will work on data curation, model training, and scalable inference across Swift-based platforms, with a focus on privacy and performance.

Join a cross-functional team building state-of-the-art language, vision, and multimodal models, collaborating with researchers and engineers to push the boundaries of AI at scale.

Qualifications

• 4+ years of experience with LLMs, LMMs, CV or related AI/ML tech.
• Experience building and evaluating ML models with data quality focus.
• Strong programming with PyTorch, TensorFlow, or JAX.
• Experience with distributed data processing frameworks.

Responsibilities

• Contribute to pre-training data curation for LLM development across domains and modalities.
• Design, implement, and deploy scalable ML models for information extraction and data generation.
• Lead small-scale research projects and collaborate on larger initiatives.
• Build data tooling to improve data velocity, reliability, and scalability.
• Advance understanding of large language models and multimodal systems.

Skills

LLMs & LMMs

Computer vision

Deep learning

Data processing

Communication skills

Education

Master's or PhD in CS/ML/AI

Tools

PyTorch

TensorFlow

JAX

Transformer models

MoE architectures

Job description

Apple in Sunnyvale, CA is seeking a senior ML engineer to help design and deploy large-scale ML systems powering Foundation Models. You will work on data curation, model training, and scalable inference across Swift-based platforms, with a focus on privacy and performance.

Join a cross-functional team building state-of-the-art language, vision, and multimodal models, collaborating with researchers and engineers to push the boundaries of AI at scale.