Internship - Machine Learning (Foundation Models)

Key Skills

Deep Learning, JAX framework, Large Language Models - LLMs, Machine learning techniques, Python Programming, PyTorch

Job Description

During your internship, you will do research and development in the context of Apple foundation models that power next-generation AI experiences across Apple’s platforms. You will learn what is relevant to transfer results from machine learning research to meaningful real-world performance at scale. Your work can contribute to Apple Intelligence and be shipped to billions of devices, worldwide.

Responsibilities

• Research and develop novel approaches for enhancing foundation model capabilities
• Leverage state-of-the-art, large-scale RL methods to improve reasoning and planning, multi-turn conversations, and general agentic capabilities.
• Design and train agents with tool calling, planning, and API integration to reliably complete tasks.
• Investigate multimodal foundation models spanning text, vision, audio, and other modalities
• Collaborate on model alignment, safety, and responsible AI practices
• Contribute to the development of tools and frameworks for foundation model experimentation and evaluation

Minimum Qualifications

• PhD student in Machine Learning (Computer Science, Electrical & Computer Engineering, Statistics, Math, Natural Sciences, or other related fields)
• Strong background in Reinforcement Learning and Deep Learning, with hands-on experience training large models, in particular LLMs
• Experience with training methodologies including pre-training, fine-tuning, and alignment approaches
• Strong programming skills. Most relevant are Python, PyTorch, and JAX

Preferred Qualifications

• Excellent problem solving, critical thinking, and interpersonal skills.
• Ability to define and drive ambitious goal-oriented research and development independently

At Apple, we're not all the same. And that's our greatest strength. We draw on the differences in who we are, what we've experienced, and how we think. Because to create products that serve everyone, we believe in including everyone. Therefore, we are committed to treating all applicants fairly and equally. We will work with applicants to make any reasonable accommodations.