AIML - Machine Learning Engineer, Applebot Crawl & Knowledge Graph Job at Apple

AIML - Sr./Staff Machine Learning Engineer, Machine Learning Platform TechnologiesSan Francisco Bay Area, California, United States Machine Learning and AIImagine what you could do here. At Apple, great ideas quickly become extraordinary products, services, and customer experiences. Bring passion and dedication to your work, and there’s no telling what you could accomplish. Do you want to make Siri and Apple products smarter for our users? The Information Intelligence teams are building groundbreaking technology for algorithmic search, machine learning, natural language processing, and artificial intelligence. The features we build are redefining how hundreds of millions of people use their computers and mobile devices to search and find what they are looking for. Within this organization, our group focuses on web-scale extraction and enrichment, transforming raw web content into structured, high-quality knowledge that powers Apple’s intelligent experiences. We design scalable extraction algorithms, develop advanced web data deduplication techniques, and apply machine learning to process trillions of records and petabytes of data. We also build and maintain Apple’s Knowledge Graph, integrating diverse data sources into a unified representation of the world knowledge. We’re looking for a Machine Learning Engineer with deep expertise in large-scale data and ML infrastructure. You will build and optimize pipelines that extract, process, and serve data artifacts while advancing the ML frameworks and tooling that underpin Apple’s knowledge and search systems.DescriptionJoin a dynamic team within Apple’s Information Intelligence Infrastructure organization that designs, builds, and operates large-scale systems powering search and AI experiences for billions of users. We develop distributed, data-intensive infrastructure that processes trillions of records and petabytes of web data, enabling large-scale extraction, enrichment, and knowledge graph construction across diverse content such as HTML, PDF, and other unstructured formats.ResponsibilitiesTackle web-scale extraction and enrichment challenges, transforming raw web data into structured knowledge that powers Siri, Spotlight, Safari and other Apple experiences.Design and optimize pipelines that leverage large language models (LLMs), advanced NLP, and entity linking frameworks to identify, deduplicate, and contextualize information from the open web.Design and evolve the infrastructure that powers large-scale extraction and enrichment pipelines, enabling Apple to transform raw web data into high-quality knowledge used across various apple products.The data and insights you help generate will directly improve Apple’s foundation models, supporting better understanding, reasoning, and grounding.Working at the intersection of web data systems, large-scale ML infrastructure, and language modeling, you will shape how Apple builds and leverages knowledge at global scale.Minimum QualificationsBachelor’s degree or higher in Computer Science or related technical field.Experience with Golang, Java or Scala.Background in computer science: algorithms, data structures, and distributed systemsExperience working in a cloud-native environment such as AWSExperience working with large-scale data processing pipelines (Spark, Cassandra, etc.)Experience with micro-service architecture in a containerized environment (Docker, Kubernetes, etc.)Experience with machine learning workflows, including feature engineering, training, evaluation, deployment and serving.Preferred QualificationsExperience with training and fine-tuning large language modelsExperience with optimizing ML training and serving performance, including GPU tuning, batch size optimization, and multi-node schedulingFamiliar with Nvidia TensorRT-LLM, vLLLM, Nvidia Triton Server etc.Excellent interpersonal skills, able to work independently as well as in a teamAt Apple, base pay is one part of our total compensation package and is determined within a range. This provides the opportunity to progress as you grow and develop within a role. The base pay range for this role is between $181,100 and $318,400, and your base pay will depend on your skills, qualifications, experience, and location.Apple employees also have the opportunity to become an Apple shareholder through participation in Apple’s discretionary employee stock programs. Apple employees are eligible for discretionary restricted stock unit awards, and can purchase Apple stock at a discount if voluntarily participating in Apple’s Employee Stock Purchase Plan. You’ll also receive benefits including: Comprehensive medical and dental coverage, retirement benefits, a range of discounted products and free services, and for formal education related to advancing your career at Apple, reimbursement for certain educational expenses — including tuition. Additionally, this role might be eligible for discretionary bonuses or commission payments as as relocation. Learn more about