Foundation ML Research Intern: RL & Agentic AI

Apple, Inc.; Zürich; On-site; CHF 22,000 - 45,000; Part time

Job summary

Apple, Inc. in Zürich, Switzerland seeks a research intern to advance foundation models with a focus on agentic capabilities and reinforcement learning. You will work with an applied ML team of about 35 engineers in Zurich and collaborate with US experts.

You will research, develop, and transfer results to real-world performance at scale, contributing to Apple Intelligence and potential product deployment across devices worldwide.

Qualifications

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

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.

Skills

Reinforcement Learning

Deep Learning

Large-Scale training

Python

PyTorch

JAX

Education

PhD student in Machine Learning

Tools

Python

PyTorch

JAX

Job description

Apple, Inc. in Zürich, Switzerland seeks a research intern to advance foundation models with a focus on agentic capabilities and reinforcement learning. You will work with an applied ML team of about 35 engineers in Zurich and collaborate with US experts.

You will research, develop, and transfer results to real-world performance at scale, contributing to Apple Intelligence and potential product deployment across devices worldwide.