Research Scientist Intern, AI & Compute Foundation - MTIA Software (PhD)

Job Description

The Meta Training & Inference Accelerator (MTIA) Software team is dedicated to developing high-performance software and hardware technologies for AI at datacenter scale, with a focus on optimizing AI production models and frameworks.

Requirements

• PhD degree in Computer Science or a related STEM field
• C/C++ programming skills
• Knowledge of Computer Architecture and Distributed systems
• Interest in High Performance Computing, Numerics, and AI hardware

Responsibilities

• Develop software stack for AI frameworks, compiler stack, and high performance kernel development
• Contribute to the optimization of PyTorch AI framework core compilers
• Analyze deep learning networks and implement compiler optimization algorithms
• Collaborate with AI research scientists on accelerating deep learning models
• Perform performance tuning and optimizations of deep learning framework components

Benefits

• Employees at Meta are often offered comprehensive benefits, including medical/dental/vision coverage, mental-health resources, family planning support (fertility, adoption, surrogacy), and caregiving programs. Financial benefits typically include a competitive retirement plan, equity awards, life insurance, and access to financial coaching and legal support. Time-off benefits commonly feature generous PTO and holidays, unlimited sick time, various paid leaves, flexible “Global Travel Days,” and a 30-day paid break every five years.