Machine Learning Engineer - Intern

Key Skills

Data science techniques, Large Language Models - LLMs, Machine learning techniques, Natural Language Processing (NLP), Python Programming, PyTorch

Job Description

As a Machine Learning (ML) Engineer, you will be entrusted with the critical role of innovating and applying state-of-the-art research in ML to tackle complex data problems. The solutions you develop will significantly impact future Apple products and the broader ML development ecosystem.

You will work with a multidisciplinary team to actively participate in the data-model co-design and co-development practice. Your responsibilities will extend to the design and development of a comprehensive data curation framework. You will also create robust model evaluation pipelines, integral to the continuous improvement and assessment of ML models. Additionally, your role will entail an in-depth analysis of collected data to underscore its influence on model performance.

Furthermore, you will have the opportunity to showcase your groundbreaking research work by publishing and presenting at premier academic venues.

Your work may span a variety of topics, including but not limited to:

• Designing and implementing semi-supervised, self-supervised representation learning techniques for maximizing the power of both limited labeled data and large-scale unlabeled data.
• Developing evaluation protocols centered on the end-to-end user experience, with a focus on anticipating potential failure modes, edge cases, and anomalies.
• Employing data selection techniques such as novelty detection, active learning, and core-set selection for diverse data types like images, 3D models, natural language, and audio.
• Uncovering patterns in data, setting performance targets, and leveraging modern statistical and ML-based methods to model data distributions. This will aid in reducing redundancy and addressing out-of-distribution samples.

Minimum Qualifications

• Currently pursuing a PhD degree or equivalent experience in Machine Learning, Computer Vision, Natural Language Processing, Data Science, Statistics or related areas.
• Proven expertise in machine learning with a passion for data-centric machine learning.
• Experience with natural language processing (NLP), and large language models, such as BERT, GPT, or Transformers.
• Strong programming skills and hands-on experience using the following languages or deep learning frameworks: Python, PyTorch, or Jax.

Preferred Qualifications

• Staying on top of emerging trends in LLMs
• Strong problem-solving and communication skills
• Demonstrated publication record in relevant conferences (e.g. NeurIPS, ICML, ICLR, CVPR, etc) is a plus
• Available for 9+ months for internship