Apple logo

AIML - ML Engineer, Batch processing team

Apple
Full-time
On-site
Seattle, Washington, United States
Imagine what you could do here. At Apple, great ideas have a way of becoming great products, services, and customer experiences very quickly. Bring passion and dedication to your job and there’s no telling what you could accomplish. Do you want to make Siri and Apple products ever smarter for our users? The Foundation Model Batch Inference team are building groundbreaking technology for large scale inference of foundation models, including state-of-the-art large language models (LLMs) and multimodal models. Our batch inference platform powers billions of foundation model inference queries across a variety of Apple products. As part of this group, you will work with one of the most exciting high performance computing environments for foundation models inference, with petabytes of data and billions of queries, and have an opportunity to imagine and build products that delight our customers every single day.

Description

We design and build infrastructures to support features that empower billions of Apple users through advanced intelligence systems. Our team processes trillions of links to find the best content to surface to users via search and other intelligent features. We also analyze pages to extract critical features for indexing, ranking, and retrieval. Join us and: * build and optimize large scale batch inference solutions * build scalable and efficient system to fully utilize diverse high performance GPU fleet

Minimum Qualifications

  • Strong coding skills
  • Strong background in computer science: algorithms and data structures
  • Strong experience with Docker containerization and Kubernetes orchestration, familiar with AWS EKS, Amazon S3 or GCP
  • Extensive expertise in designing robust, large scale backend system, taking into account for performance, scalability, security and maintainability.
  • Excellent interpersonal skills able to work independently as well as in a team

Key Qualifications

Preferred Qualifications

  • Familiar with one of the popular ML Frameworks like PyTorch, Tensorflow
  • Familiar with foundation model architectures such as Transformers, Encoder/Decoder
  • Familiar with fundamental model inference
  • Familiar with MapReduce style batch jobs

Education & Experience

Additional Requirements

Pay & Benefits

  • Apple is an equal opportunity employer that is committed to inclusion and diversity. We take affirmative action to ensure equal opportunity for all applicants without regard to race, color, religion, sex, sexual orientation, gender identity, national origin, disability, Veteran status, or other legally protected characteristics. Learn more about your EEO rights as an applicant.