Research Scientist Intern, Foundation Model Optimization (PhD)

Meta; Sunnyvale (CA); On-site; USD 50,000 - 75,000; Full time

Job summary

Meta is seeking a Research Scientist Intern to advance language and vision model optimization for AR/VR/SG devices within Reality Labs. You will work on cutting-edge efficient LLMs and foundation models, focusing on deployment in mobile and mixed-reality environments.

The role emphasizes collaboration across device and cloud, personalization, and model compression, with opportunities to publish in top conferences as part of a multidisciplinary team.

Qualifications

• PhD student in AI or related field.
• Must have work authorization at the time of hire and maintain it.
• Hands-on experience with deep learning algorithms and techniques (e.g., transformers).
• Experience with PyTorch/JAX and Python/shell scripting.

Responsibilities

• Perform LLM and vision foundation model research that leads to publications.
• Develop model optimization algorithms and infrastructure for resource-constrained deployment.
• Research joint optimization of models and the surrounding assistant system across device and cloud.
• Explore personalization, model compression and efficient inference, omni-modal models.
• Collaborate with cross-disciplinary research and engineering teams.

Skills

Deep learning

Transformer models

Model optimization

On-device inference

Education

PhD in Artificial Intelligence or related field

Tools

PyTorch

JAX

Python

Shell scripting

Job description

Meta's mission is to give people the power to build community and bring the world closer together, and the Meta Reality Labs organization focuses on delivering this vision through AR/VR/SG devices.
Meta's Efficiency and On-Device AI team within the Reality Labs organization is seeking a Research Scientist Intern with a focus on language/vision model optimization. As a Research Scientist Intern, your primary objective is to develop cutting-edge efficient LLMs and vision foundation models.

Responsibilities

• Perform LLM and vision foundation model research that leads to publications in top-tier CV/NLP/ML conferences.
• Develop model optimization algorithms and infrastructure to enable deployment of models in resource-constrained settings, such as mobile phones and AR/VR/MR/SG devices.
• Research joint optimization of models and the surrounding assistant system, including planning, memory, tool use, and coordination across device and cloud.
• Explore personalization, model compression and efficiency, and novel model architectures including omni-modal capabilities.
• Collaborate with cross-disciplinary research and engineering teams.

Minimum qualifications

• Currently is in the process of obtaining a PhD in the field of Artificial Intelligence or related field
• Must obtain work authorization in the country of employment at the time of hire, and maintain ongoing work authorization during employment
• Hands-on experience in deep learning algorithms and techniques, e.g., transformers
• Experience with deep learning software libraries such as PyTorch/JAX
• Experience with scripting languages such as Python and shell scripts

Preferred qualifications

• Intent to return to degree-program after the completion of the internship
• Experience with large scale foundation model training and on-device foundation model optimization
• Experience in one or more of: LLM agents and orchestration, personalization, model compression and efficient inference, efficient model architectures, multi-modal modeling, data-centric approaches for model tuning and evaluation
• Proven track record of publications at CV/NLP/ML conferences/journals such as NeurIPS, ICLR, ICML, CVPR, ICCV, ECCV, AAAI, ACL, EMNLP