Job was saved successfully.
Job was removed from Saved Jobs.

Job Details

Meta (formerly known as Facebook)

Research Scientist Intern, On-device ML Accelerator Compiler (PhD)

Computer and Mathematical



Redmond, Washington, United States

Meta AR/VR focuses on delivering Meta’s vision through Virtual Reality (VR) and Augmented Reality (AR). Enabling compelling user experiences on Virtual and Augmented Reality devices requires innovation and co-design across all layers of stack from novel algorithms to custom silicon. Meta’s AR/VR team is driving the state of the art forward with breakthrough work in computer vision, speech, virtual assistant, machine learning, mixed reality, graphics, displays, sensors, and new ways to map the human body among many others.
We are seeking exceptional interns with a background in developing compilers and/or programming languages. Our compiler targets on-device ML accelerators for mobile AR/VR applications, including VR headsets and AR glasses.
You would join a team developing tooling for automatic PyTorch program capture and compilation for one of Meta’s machine learning accelerators. You will collaborate with teams working on the development stack from model development to compilation, via Torch FX transformations.

Required Skills

Research Scientist Intern, On-device ML Accelerator Compiler (PhD) Responsibilities:

  • Design and implement extensions to existing PyTorch model capture tooling inside the framework of an existing on-device ML accelerator compiler pipeline
  • Collaborate with other research scientists and software engineers to propose changes to both models and tooling in a co-design effort
  • Capture, create and execute test cases to show the increased scope of program capture
  • Communicate design ideas and results clearly, both within the group as well as to the cross-functional groups
  • Note: no prior experience or knowledge of Torch FX is expected. You will be inducted on Torch FX as part of the internship.

Minumum Qualification

Minimum Qualifications:

  • Must obtain work authorization in the country of employment at the time of hire and maintain ongoing work authorization during employment
  • Currently has, or is in the process of obtaining, a PhD degree in Computer Science, Electrical Engineering or related field
  • 2+ years experience in developing compilers or programming language design
  • 2+ years programming experience
  • Working knowledge of the Python programming language
  • Interpersonal experience: cross-group and cross-culture collaboration

Preferred Qualification

Preferred Qualifications:

  • Experience with the PyTorch deep learning framework
  • Experience in program static analysis or program capture
  • Basic knowledge of machine learning models
  • Intent to return to degree-program after the completion of the internship/co-op
  • Publication track record in machine learning conferences and/or journals

Meta is proud to be an Equal Employment Opportunity and Affirmative Action employer. We do not discriminate based upon race, religion, color, national origin, sex (including pregnancy, childbirth, reproductive health decisions, or related medical conditions), sexual orientation, gender identity, gender expression, age, status as a protected veteran, status as an individual with a disability, genetic information, political views or activity, or other applicable legally protected characteristics. You may view our Equal Employment Opportunity notice [Register to View] . We also consider qualified applicants with criminal histories, consistent with applicable federal, state and local law. We may use your information to maintain the safety and security of Meta, its employees, and others as required or permitted by law. You may view [Register to View] , [Register to View] notice, and [Register to View] by clicking on their corresponding links. Additionally, Meta participates in the [Register to View] in certain locations, as required by law