Lead ML Engineer - Generative AI

Key Skills

Aartificial intelligence, Apache Airflow, BERT, ChatGPT, Docker, Effective communication skills, Generative AI, Git, GPUs, High performance computing- HPC, JAX framework, Kubernetes-K8s, LangChain, Machine learning techniques, MLflow, MLOps tools, PyTorch, TensorFlow

Job Description

Imagine what you could do here. At Apple, we believe new insights have a way of becoming excellent products, services, and customer experiences very quickly. Bring passion and dedication to your job and there's no telling what you could accomplish.

The people here at Apple don’t just create products — they create the kind of wonder that’s revolutionized entire industries. It’s the diversity of those people and their ideas that inspires the innovation that runs through everything we do, from amazing technology to industry-leading environmental efforts. Join Apple, and help us leave the world better than we found it.

It takes deeply dedicated, intelligent individuals to maintain and exceed the high expectations at Apple. Cloud Technologies App Platform team is looking for an extraordinary Machine Learning engineer to join our team. You will help design and implement our machine learning strategy to take our developer experience platform to the next level and help accelerate app development inside Apple. We will be collaborating and working with multi-functional teams and applying algorithms to large-scale data.

Key Qualifications

• Solid understanding and experience in advanced machine learning algorithms, deep learning, statistics, and large language models.
• Demonstrable experience with transformer models such as BERT, GPT, RoBERTa, etc, and a solid understanding of their underlying principles is a plus
• Model optimization and fine tuning of large language models for better performance, latency, memory and throughput
• Strong software development skills with proficiency in Python, PyTorch, TensorFlow, JAX
• Building MLOps infrastructure for experimentation and A/B testing and production deployment and operations.
• Experience with MLOps technology: Git, Docker, Kubernetes, MLFlow/DVC/ClearML, Airflow
• Setting up LangChain pipeline including prompt engineering and parsing
• Ability to meaningfully present results of analyses in a clear and impactful manner. Able to break down sophisticated machine learning model knowledges to digestible information for non-technical audience.
• Experience in applying ML techniques in developer productivity is a plus
• 5+ years of overall experience in the software industry focussed on AI/ML

App Platform team provides frameworks, services and developer tools to internal engineering teams in Apple to accelerate their SDLC. We are working on a Generative AI based developer experience platform for internal use and assist our app development team. You will work with data scientists, software engineers and operations to deliver and end to end ML enabled solution for this platform.

You will work with data scientists to deploy and evaluate different models and fine tune it to custom needs. You will also explore-many integrations using tools like LangChain, IDE plugins, git based hooks to innovate on ways to improve developer experience.

Education & Experience

B.Tech/M.Tech/M.S. in related field or Ph.D. in Data Science, Machine Learning, Operations Research with 5+ years experience