← All roles

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

Meta· Meta AI / MTIA Software
Apply on company site
Menlo Park, CA$7.7k/mo - $12.7k/moInternPosted 1w ago

About the role

Internship in MTIA Software focusing on AI acceleration software and hardware co-design for datacenter scale. Work on AI frameworks, compiler stacks, and high-performance kernel development to accelerate Meta's DL/ML workloads on MTIA AI accelerator hardware.

Responsibilities

  • Develop SW stack with core focus areas: AI frameworks, compiler stack, high performance kernel development and acceleration onto next generation hardware architectures.
  • Contribute to development of the PyTorch AI framework core compilers to support new state of the art inference and training AI hardware accelerators and optimize their performance.
  • Analyze deep learning networks, develop & implement compiler optimization algorithms.
  • Collaborate with AI research scientists to accelerate next generation of deep learning models such as Recommendation systems, Generative AI, Computer vision, NLP etc.
  • Performance tuning and optimizations of deep learning framework & software components.

Qualifications

  • In the process of obtaining PhD degree in Computer Science or related STEM field
  • Research experiences in AI compilers, high-performance kernel development, deep learning framework and runtime internals, numerics and low-precision formats, or parallelism strategies for large-scale training and inference
  • Knowledge of machine learning systems and computer architecture, including memory and storage hierarchy, parallel execution, and performance analysis on accelerated hardware
  • Proven C/C++, Python or Rust programming skills
  • Must obtain work authorization in the country of employment at the time of hire, and maintain ongoing work authorization during employment

Skills

Required:PhD in Computer Science or related STEM fieldAI compilershigh-performance kernel developmentdeep learning framework and runtime internalsC/C++Python
Nice to have:numerics and low-precision formatsparallelism strategies for large-scale training/inferenceRust

Benefits

  • Benefits offered by Meta (explicit benefits not enumerated in job description)
Last updated Oct 23, 2026. Interested? Apply on the company site.

About Meta

Shop the latest AI glasses and wearable technology from Meta.

Shop the latest AI glasses and wearable technology from Meta. Explore Ray-Ban Meta, Oakley Meta, display glasses, and cutting-edge VR headsets.

AI glassesOakley MetaRay-Ban MetaVR headsetsliquid-ai-recruiting-targetwearable technology
Company profileWebsite

More open roles at Meta