Machine Learning Engineer

What you'll do

• Build state-of-the-art Generative AI components at scale to power Siri's question answering features.
• Develop machine learning systems that process and manage large-scale graph data.
• Perform knowledge extraction from both semi-structured and unstructured data feeds.
• Design and develop features for a high-performance platform handling millions of queries per second.
• Manage petabytes of data using large-scale data pipelines and the Apache software stack.
• Implement NLP solutions utilizing Large Language Models and RAG-based architectures.
• Develop production-ready machine learning models using frameworks like PyTorch or TensorFlow.

What we're looking for

• Excellent programming skills in mainstream languages such as Python, Scala, and Go.
• Prior experience with Machine Learning Frameworks like PyTorch and TensorFlow.
• Working experience with Natural Language Processing (NLP) and Large Language Models (LLMs).
• Experience working with large datasets and pipelines, preferably using the Apache software stack like Spark.
• Broad knowledge of computer science and systems with an ability to learn new domains quickly.
• Excellent problem-solving, analytical, and interpersonal skills.
• A PhD or Graduate degree in Computer Science, AI, Machine Learning, Information Retrieval, Data Science, or a related field.
• Experience in NLP including RAG based Generative AI and transformer architecture.