Logo of Huzzle

Home

Matches

Explore

Manage

Research Scientist Intern, PhD, PyTorch Model Optimization

image

Meta

Feb 12

Applications are closed

  • Internship
    Full-time
    Off-cycle Internship
  • Research & Development
    Software Engineering
  • San Francisco

Requirements

  • Currently has, or is in the process of obtaining, PhD degree in the field of Computer Science or a related STEM field
  • Experience in one or more of the following machine learning/deep learning domains: Running fine-tuning and inference/evaluation on AI models, ML theory: Basic knowledge about ML models in different modalities like LLM (Large Language Models), Vision (VITS, MVITS) and Multimodal and how scale impacts performance, ML systems: AI infrastructure, machine learning accelerators, high performance computing, machine learning compilers, GPU architecture, machine learning frameworks, distributed systems, on-device optimization.
  • Must obtain work authorization in the country of employment at the time of hire, and maintain ongoing work authorization during employment
  • Preferred Qualifications:
  • Experience or knowledge on training models at scale using PyTorch/TensorFlow/JAX.
  • Experience or knowledge on working with a distributed GPU cluster.
  • Publications in top tier ML or System Conferences such as ASPLOS, ICML, ICLR, KDD, NeurIPS, MLSys, SOSP, OSDI, NSDI.
  • Intent to return to degree-program after the completion of the internship/co-op

Responsibilities

  • Apply relevant AI and machine learning techniques to advance the state-of-the-art in machine learning frameworks.
  • Collaborate with users of PyTorch to enable new use cases for the framework both inside and outside Meta.
  • Develop novel, accurate optimization algorithms for AI models during fine-tuning or inference.

Meta

Technology
Industry
10,001+
Employees
2004
Founded Year

Mission & Purpose

Meta builds technologies that help people connect, find communities, and grow businesses. When Facebook launched in 2004, it changed the way people connect. Apps like Messenger, Instagram and WhatsApp further empowered billions around the world. Now, Meta is moving beyond 2D screens toward immersive experiences like augmented and virtual reality to help build the next evolution in social technology.