AIML - Sr. Machine Learning Engineer - ML Platform & Technology (MLPT)

Apple - Santa Clara, CA

Full-time - Senior

Santa Clara, CA

Computer and Electronic Product Manufacturing

About the position

The Sr. Machine Learning Engineer for the ML Platform & Technology (MLPT) at Apple will play a crucial role in advancing intelligent experiences across Apple's products and services. This position focuses on leveraging the latest advancements in Generative AI and Large Language Models to enhance machine learning infrastructure and enable faster iteration for ML teams. The engineer will collaborate with various teams to optimize the platform and deliver innovative solutions to complex ML challenges.

Responsibilities

• Enable ML teams to iterate their projects faster through improved infrastructure and tools.
• Define and build new features to address challenges in generative AI and large language models.
• Collaborate with internal ML teams to adopt and apply the platform effectively.
• Gather feedback from partners to enhance user experience and scalability of products and services.

Requirements

• A minimum of 3 years as a tech lead.
• 8 years of proven experience in machine learning related software engineering, including product, infrastructure, frameworks, or tools.
• Strong expertise in machine learning domains such as computer vision (CV), natural language processing (NLP), recommender systems, Generative AI, or large language models (LLMs).
• Proficient in ML training and deployment frameworks like TensorFlow, PyTorch, Faster Transformer, TensorRT, and vLLM.
• Proficient in cloud computing and data processing infrastructure and tools such as Kubernetes, Ray, PySpark, and SQL.
• Ability to communicate technical and architectural problems clearly and concisely.

Nice-to-haves

• Advanced degree in a related field.
• Additional expertise in machine learning domains such as CV, NLP, recommender systems, Generative AI, or LLMs.