SWE - Machine Learning Researcher, Foundation Models

Who you are

You have a PhD or equivalent practical experience in Computer Science or a related technical field, demonstrating expertise in deep learning with a solid publication record in relevant conferences such as NeurIPS, ICML, or ACL. Your programming skills in Python are complemented by proficiency in deep learning toolkits like PyTorch or TensorFlow, allowing you to effectively contribute to innovative projects. You thrive in collaborative environments, working closely with engineers and scientists to tackle complex challenges in foundation models and deep learning. Your passion for developing algorithms and techniques drives you to push the boundaries of what is possible in AI applications.

Desirable

Experience with additional deep learning frameworks such as JAX or Triton will be a plus, as will any background in applying deep learning techniques to real-world products. You are eager to identify and develop novel applications of deep learning that can significantly impact user experiences.

What you'll do

In this role, you will be part of a dedicated team responsible for building infrastructure, datasets, and models that enhance Apple's product offerings. You will engage in research that applies deep learning to real-world use cases, aiming to solve some of the most interesting problems in the field. Your work will involve developing algorithms and systems that leverage foundation models to improve the understanding and generation of text, images, speech, and videos. You will collaborate with a close-knit team of world-class engineers and scientists, contributing to projects that delight millions of users with innovative AI solutions. Your ideas will have a direct impact on the user experience, as you work on cutting-edge technologies that push the frontier of deep learning.

What we offer

At Apple, you will be part of a culture that values innovation and collaboration. You will have the opportunity to work on projects that are at the forefront of technology, impacting billions of users worldwide. We encourage you to apply even if your experience doesn't match every requirement, as we believe diverse teams build better products. Join us in our mission to create groundbreaking AI solutions that enhance the lives of our users.