Experience Inc. Jobs

Job Information

Facebook Research Scientist, PyTorch Compiler (University Grad) in Denver, Colorado


Facebook's mission is to give people the power to build community and bring the world closer together. Through our family of apps and services, we're building a different kind of company that connects billions of people around the world, gives them ways to share what matters most to them, and helps bring people closer together. Whether we're creating new products or helping a small business expand its reach, people at Facebook are builders at heart. Our global teams are constantly iterating, solving problems, and working together to empower people around the world to build community and connect in meaningful ways. Together, we can help people build stronger communities - we're just getting started.


Our team’s mission is to make PyTorch faster and easier to use in order to create and maintain a state-of-the-art machine learning framework that is used across Facebook and the entire industry. The key challenges in the team are extracting structure from PyTorch programs (via TorchScript, torch.fx, and Lazy Tensors), generating fast CPU/GPU code (via runtimes, graph optimizations, and code generation), and enabling deployment of PyTorch (via hardware accelerators and embedding PyTorch inside other systems).

Required Skills:

  1. Develop new techniques to improve the PyTorch compiler.

  2. Find opportunities for new optimizations to make PyTorch faster.

  3. Apply knowledge and research to advance the state-of-the-art in machine learning frameworks.

  4. Collaborate with users of PyTorch to enable new use cases for the framework both inside and outside Facebook.

Minimum Qualifications:

  1. Currently has or in the process of obtaining PhD or BS/MS in computer science or related field and 2+ years of experience.

  2. Experience in machine learning, systems, programming languages, or compilers.

  3. Must obtain work authorization in country of employment at the time of hire and maintain ongoing work authorization during employment.

Preferred Qualifications:

  1. Research and software engineer experience demonstrated via grants, fellowships, patents, internship, work experience, and/or coding competitions.

  2. Experience doing performance optimization on machine learning models.

  3. First-authored publications at peer-reviewed conferences (e.g. NeurIPS, MLSys, ASPLOS, PLDI, CGO, PACT, ICML, or similar).

Industry: Internet

Equal Opportunity: Facebook is proud to be an Equal Opportunity and Affirmative Action employer. We do not discriminate based upon race, religion, color, national origin, sex (including pregnancy, childbirth, or related medical conditions), sexual orientation, gender, gender identity, gender expression, transgender status, sexual stereotypes, age, status as a protected veteran, status as an individual with a disability, or other applicable legally protected characteristics. We also consider qualified applicants with criminal histories, consistent with applicable federal, state and local law. Facebook is committed to providing reasonable accommodations for candidates with disabilities in our recruiting process. If you need any assistance or accommodations due to a disability, please let us know at accommodations-ext@fb.com.