Foundation Models Data Research Engineer

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

Benefits offered by this job

Apple Benefits

Relocation assistance

Job summary

Apple Inc. in Cupertino, CA is seeking a Research Engineer for Pretraining Data on Foundation Models. You will own ambitious projects spanning research, algorithm development, experimentation, and production engineering to improve data quality, diversity, and efficiency for next-generation models.

You will collaborate with model researchers to run controlled experiments and translate ideas into scalable data pipelines, contributing to the advancement of foundation models powering Apple products.

Qualifications

• Advanced degree or equivalent practical experience in CS/ML.
• Strong Python and production-grade software development skills.
• Experience with large-scale data processing and ML systems.

Responsibilities

• Design and implement algorithms for large-scale pretraining data curation.
• Build scalable data processing systems for trillions of tokens.
• Collaborate with researchers to design training experiments and analyze results.
• Translate research prototypes into robust production data systems.
• Stay current with foundation model research and apply new techniques.

Skills

Python programming

Distributed computing

Machine learning fundamentals

Problem solving

Education

BS/MS/PhD in CS/ML or related

Tools

PyTorch

Spark

Ray

Job description

Apple Inc. in Cupertino, CA is seeking a Research Engineer for Pretraining Data on Foundation Models. You will own ambitious projects spanning research, algorithm development, experimentation, and production engineering to improve data quality, diversity, and efficiency for next-generation models.

You will collaborate with model researchers to run controlled experiments and translate ideas into scalable data pipelines, contributing to the advancement of foundation models powering Apple products.