Staff / Sr. Machine Learning Engineer, AI, Search & Knowledge Platforms

• Bachelor’s degree or higher in Computer Science or related technical field
• 3+ years of experience in software engineering or ML engineering
• Experience with Golang, Java, Scala, or Python
• Background in computer science: algorithms, data structures, and distributed systems
• Experience working in a cloud-native environment such as AWS
• Experience 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, * Experience with training and fine-tuning large language models
• Experience with optimizing ML training and serving performance, including GPU tuning, batch size optimization, and multi-node scheduling
• Familiarity with Nvidia TensorRT-LLM, vLLM, Nvidia Triton Server, or similar inference frameworks.
• Experience with NLP, information extraction, or web data systems.
• Excellent interpersonal skills, able to work independently as well as in a team

Join 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 web data at global scale, enabling extraction, enrichment, and knowledge graph construction across diverse content such as HTML, PDF, and other unstructured formats.