Machine Learning Engineer

About the job

As a Machine Learning Engineer, you will work on building intelligent systems to democratize AI across a wide range of solutions within Apple. You will drive the development and deployment of state-of-the-art AI models and systems that directly impact the capabilities and performance of Apple’s products and services. You will implement robust, scalable ML infrastructure, including data storage, processing, and model serving components, to support seamless integration of AI/ML models into production environments. You will develop novel feature engineering, data augmentation, prompt engineering and fine-tuning frameworks that achieve optimal performance on specific tasks and domains. You will design and implement automated ML pipelines for data preprocessing, feature engineering, model training, hyper-parameter tuning, and model evaluation, enabling rapid experimentation and iteration. You will also implement advanced model compression and optimization techniques to reduce the resource footprint of language models while preserving their performance. Have continuous focus to Brainstorm and Design various POCs using AI/ML Services for new or existing enterprise problems. Minimum Qualifications• Bachelor’s Degree or Equivalent with 5+ years of experience in AI/ML & Data engineering• Ability to design systems that go from raw data → structured insight → autonomous action, closing the loop between data engineering and AI decision-making• Design and implement RAG (Retrieval-Augmented Generation) pipelines that feed structured data, user personas, and domain-specific context into LLMs for accurate, grounded outputs• Deep understanding of embeddings, vector databases, and semantic search to power retrieval layers in AI applications• Hands-on experience building AI agents with memory, planning, tool use, and execution loops using frameworks such as LangChain, LlamaIndex.• Proven expertise with transformer-based models (BERT, GPT, LLaMA, etc.) including fine-tuning, prompt engineering, and understanding of their underlying attention mechanism• Proficiency with ML frameworks such as PyTorch or TensorFlow, including model training, optimization, and serving pipelines• Experience in modern cloud warehouse, data lakes and implementation experience on any of the cloud platforms like AWS/GCP/Azure - preferably AWS.• Experience working with data at scale (peta bytes) with big data tech stack and advanced programming languages e:g Python, Scala.

Requirements

• AI/ML
• Data Engineering
• Machine Learning Frameworks
• Python
• Pytorch
• TensorFlow

Qualifications

• Bachelor’s Degree or Equivalent
• 5+ years of experience in AI/ML & Data engineering

Preferred Technologies

• AI/ML
• Data Engineering
• Machine Learning Frameworks
• Python
• Pytorch
• TensorFlow

About the company

The people here at Apple don’t just build products — we craft the kind of wonder that’s revolutionized entire industries. It’s the diversity of those people and their ideas that supports the innovation that runs through everything we do, from amazing technology to industry-leading environmental efforts.